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

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

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

 

小兵   2019-01-30 15:44



   1个回答 
2

用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面经 问答 实战

wxw_pku   2019-01-30 16:15



  相关主题

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

sklearn.linear_model.SGDClassifier中的参数n_iter设置问题   2回答

查看sklearn版本   1回答

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

怎么自定义sklearn GridSearchCV中评估函数的阈值   1回答

sklearn训练classifier的时候报错Unknown label type   1回答

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

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

sklearn可以用gpu加速吗?   1回答

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

GradientBoostingClassifier的每次输出结果不同   1回答

sklearn predict的默认阈值   2回答



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

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

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

我们谢绝答非所问。

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

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