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

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

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

 

小兵   2019-01-30 15:44



   1个回答 
5

用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.linear_model.SGDClassifier能做小批量学习(mini-batch)吗?   2回答

sklearn里learning_rate什么意思?   2回答

sklearn中的Randomizedlogisticregression是什么算法?和SGDClassifier一样吗?   2回答

CatBoostClassifier里的rsm参量是什么意思?   1回答

提取sklearn逻辑回归的模型系数   1回答

sklearn中除了sgd以外,还有什么可以使用partial_fit方法吗?   1回答

怎么获取sklearn.linear_model.SGDRegressor的回归系数、截距?   1回答

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

查看sklearn版本   1回答

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

sklearn predict的默认阈值   2回答

sklearn里的LabelEncoder什么用?   3回答



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

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

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

我们谢绝答非所问。

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

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