我可以对sklearn中的predict_proba方法给出的结果设置更高的精度吗? 感谢
答案 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
,您可以为新的未见过的样本选择合适的阈值,以反映您所需的精度