增加sklearn中的predict_proba精度

时间:2015-02-22 22:27:08

标签: scikit-learn classification precision

我可以对sklearn中的predict_proba方法给出的结果设置更高的精度吗? 感谢

2 个答案:

答案 0 :(得分:1)

predict_proba返回的概率为float

prediction_probs = classifier.predict_proba(a_test) 

type(prediction_probs[1][1])

返回:

numpy.float64

prediction_probs[1][1]

返回:

0.40000000000000002

答案 1 :(得分:0)

您可以在 predict_proba 之后放置一个条件,以便仅当概率超过您的自定义阈值时才预测样本为正

custom_threshold = .9
y_test_pred = classifier.predict_proba(x_test) > custom_threshold

通过查看测试集上的 sklearn.metrics.precision_recall_curve,您可以为新的未见过的样本选择合适的阈值,以反映您所需的精度