提高sklearn中的predict_proba精度



我可以对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,您可以为新的看不见的样本选择一个合适的阈值,以便它反映您想要的精度

相关内容

  • 没有找到相关文章

最新更新