我正在使用sklearn
进行回归,并使用随机网格搜索来评估不同的参数。这是一个玩具示例:
from sklearn.datasets import make_regression
from sklearn.metrics import mean_squared_error, make_scorer
from scipy.stats import randint as sp_randint
from sklearn.ensemble import ExtraTreesRegressor
from sklearn.cross_validation import LeaveOneOut
from sklearn.grid_search import GridSearchCV, RandomizedSearchCV
X, y = make_regression(n_samples=10,
n_features=10,
n_informative=3,
random_state=0,
shuffle=False)
clf = ExtraTreesRegressor(random_state=12)
param_dist = {"n_estimators": [5, 10],
"max_depth": [3, None],
"max_features": sp_randint(1, 11),
"min_samples_split": sp_randint(1, 11),
"min_samples_leaf": sp_randint(1, 11),
"bootstrap": [True, False]}
rmse = make_scorer(mean_squared_error, greater_is_better=False)
r = RandomizedSearchCV(clf, param_distributions=param_dist,
cv=10,
scoring='mean_squared_error',
n_iter=3,
n_jobs=2)
r.fit(X, y)
我的问题是:
1)RandomizedSearchCV
使用r2
作为评分函数吗?没有记录归档的默认评分函数。
2)即使我在代码中使用mean_squared_error
作为评分函数,为什么分数为负(如下所示)? mean_squared_error
应该都是积极的。然后当我计算r.score(X,y)
时,似乎再次报告R2
。在所有这些背景下的分数对我来说非常混乱。
In [677]: r.grid_scores_
Out[677]:
[mean: -35.18642, std: 13.81538, params: {'bootstrap': True, 'min_samples_leaf': 9, 'n_estimators': 5, 'min_samples_split': 3, 'max_features': 3, 'max_depth': 3},
mean: -15.07619, std: 6.77384, params: {'bootstrap': False, 'min_samples_leaf': 7, 'n_estimators': 10, 'min_samples_split': 10, 'max_features': 10, 'max_depth': None},
mean: -17.91087, std: 8.97279, params: {'bootstrap': True, 'min_samples_leaf': 7, 'n_estimators': 10, 'min_samples_split': 7, 'max_features': 7, 'max_depth': None}]
In [678]: r.grid_scores_[0].cv_validation_scores
Out[678]:
array([-37.74058826, -26.73444271, -36.15443525, -23.11874605,
-33.60726519, -33.4821689 , -36.14897322, -43.80499446,
-68.50480995, -12.97342433])
In [680]: r.score(X,y)
Out[680]: 0.87989839693054017
答案 0 :(得分:3)
就像GridSearchCV一样,RandomizedSearchCV默认使用估算器上的score
方法。 ExtraTreesRegressor和其他回归估算器从此方法返回R²分数(分类器返回准确度)。
惯例是分数是最大化的。均方误差是一种最小化的损失函数,所以它在搜索中被否定了。
然后当我计算r.score(X,y)时,似乎再次报告R2。
那不漂亮。这可能是一个错误。