【发布时间】:2021-09-29 04:27:11
【问题描述】:
我正在研究一个二元分类问题。我有这种情况,我使用了从 sklearn 导入的逻辑回归和支持向量机模型。这两个模型拟合相同,不平衡的训练数据和类权重进行了调整。他们取得了可比的表现。当我使用这两个预训练模型来预测新数据集时。 LR 模型和 SVM 模型将相似数量的实例预测为正例。并且预测的实例有很大的重叠。
但是,当我查看被分类为阳性的概率分数时,LR 的分布是从 0.5 到 1,而 SVM 从大约 0.1 开始。我调用了函数model.predict(prediction_data)来找出预测为每个类的实例和函数
model.predict_proba(prediction_data) 给出被分类为 0(neg) 和 1(pos) 的概率分数,并假设它们都有一个默认阈值 0.5。
我的代码没有错误,我不知道为什么 SVM 将概率分数
【问题讨论】:
标签: machine-learning scikit-learn svm prediction