我可以对sklearn中的方法predict_proba给出的结果设置更高的精度吗?感谢
predict_proba
返回的概率为float
:
prediction_probs = classifier.predict_proba(a_test)
type(prediction_probs[1][1])
退货:
numpy.float64
和
prediction_probs[1][1]
退货:
0.40000000000000002
您可以在predict_proba之后设置一个条件,以便只有当概率超过您的自定义阈值时,样本才会被预测为阳性
custom_threshold = .9
y_test_pred = classifier.predict_proba(x_test) > custom_threshold
通过查看测试集上的sklearn.metrics.precision_recall_curve
,您可以为新的看不见的样本选择一个合适的阈值,以便它反映您想要的精度