如何在sklearn网格搜索中使用日志丢失

时间:2016-03-18 16:05:57

标签: python python-2.7 scikit-learn

我正在尝试使用日志丢失的网格搜索。以下是我的代码。

df = pd.read_csv(file_name)
df.shape # (146614, 395)
# All features are float32
gbm = GradientBoostingClassifier(max_features="log2")
GSCV = GridSearchCV(gbm, param, scoring=log_loss, n_jobs=2, cv =2, verbose=3)
GSCV.fit(df, y)
clf.grid_scores_

但是,我无法运行代码。在GSCV.fit线上很长一段时间后它给我错误。但是,当我放弃得分= log_loss时它工作正常。

有人可以给我建议吗?

2 个答案:

答案 0 :(得分:10)

在sklearn中尝试scoring='neg_log_loss'> 0.18

答案 1 :(得分:1)

您可以使用" log_loss" 作为字符串而不是函数。