sklearn分类模型中预测结果怎么调整概率的阈值

  统计/机器学习 监督式学习 Python    浏览次数:526        分享
0

请问各位,在分类算法中,预测的结果,如果概率>50%为1,概率小于<50% 为0 ,请问在sklearn中,怎么调整这个阈值,具体的程序怎么写,谢谢!!

 

小兵   2019-01-30 15:44



   1个回答 
4

用predict_proba得到概率,然后对概率切一下。以LogisticRegression为例

clf = LogisticRegression()
clf.fit(trainX, ytrain)
pred_proba = clf.predict_proba(testX)[:, 1]
thres = 0.75
pred_label = pred_proba > thres

pred_proba是每个点为真的概率

假设阈值是0.75

pred_label里True就是概率大于0.75的

SofaSofa数据科学社区DS面试题库 DS面经

wxw_pku   2019-01-30 16:15



  相关主题

sklearn predict的默认阈值   2回答

lightgbm.LGBMModel和lightgbm.LGBMClassifier该用哪个?   1回答

sklearn SGDClassifier的partial_fit是什么意思?   2回答

sklearn的模型如何保存下来?   2回答

sklearn里的LabelEncoder什么用?   3回答

关于sklearn.model_selection.PredefinedSplit的用法   1回答

sklearn实现adaboost算法了吗?   1回答

sklearn classification_report里的support是什么意思   1回答

sklearn里learning_rate什么意思?   2回答

sklearn中的predict_proba方法的返回值的意义   2回答

sklearn.linear_model.SGDClassifier能做小批量学习(mini-batch)吗?   2回答

sklearn回归的入门级问题   3回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!