请问各位，在分类算法中，预测的结果，如果概率>50%为1，概率小于<50% 为0 ，请问在sklearn中，怎么调整这个阈值，具体的程序怎么写，谢谢！！
1个回答
用predict_proba得到概率，然后对概率切一下。以LogisticRegression为例
clf = LogisticRegression()
clf.fit(trainX, ytrain)
pred_proba = clf.predict_proba(testX)[:, 1]
thres = 0.75
pred_label = pred_proba > threspred_proba是每个点为真的概率
假设阈值是0.75
pred_label里True就是概率大于0.75的
SofaSofa数据科学社区DS面试题库 DS面经
wxw_pku
2019-01-30 16:15
相关讨论
CatBoostClassifier里的rsm参量是什么意思？
sklearn.linear_model.SGDClassifier能做小批量学习(mini-batch)吗？
sklearn.linear_model.SGDClassifier中的参数n_iter设置问题
sklearn中的Randomizedlogisticregression是什么算法？和SGDClassifier一样吗？
sklearn中的predict_proba方法的返回值的意义
随便看看