怎么自定义sklearn GridSearchCV中评估函数的阈值,比如说我想把precision定义为大于0.7才算是true
1个回答
可能你需要用make_scorer自己定义一个scoring了,然后在GridSearchCV把scoring=你的scoring函数名
-----二更-----
也可以不用make_scorer,自己定义一个,在自己定义的scoring里面调用predict_proba得到概率
下面是一个例子
def my_scorer(clf, X, y_true):
class_labels = clf.classes_
y_pred_proba = clf.predict_proba(X)
error = ...
return error
gs = GridSearchCV(estimator=KNeighborsClassifier(),
param_grid=[{'n_neighbors': [6]}],
cv=5,
scoring=my_scorer)
但是传入到自定义scoring里面的参数是预测结果0或1,我怎么把这个预测值变成概率?
-
quentin
2018-11-28 15:59
我更新了我的回答,可以看看
-
木子周
2018-11-28 16:23
成立,十分感谢
-
quentin
2018-12-05 13:31