python predict_proba_Scikit learn predict_proba给出错误答案

在这个问题上,我引用了以下代码:>>> import sklearn

>>> sklearn.__version__

'0.13.1'

>>> from sklearn import svm

>>> model = svm.SVC(probability=True)

>>> X = [[1,2,3], [2,3,4]] # feature vectors

>>> Y = ['apple', 'orange'] # classes

>>> model.fit(X, Y)

>>> model.predict_proba([1,2,3])

array([[ 0.39097541, 0.60902459]])

我在那个问题中发现这个结果代表了属于每个类的点的概率,按照model.classes给出的顺序>>> zip(model.classes_, model.predict_proba([1,2,3])[0])

[('apple', 0.39097541289393828), ('orange', 0.60902458710606167)]

所以。。。这个答案,如果解释正确的话,可能是一个‘橙色’(由于数据量很小,可信度相当低)。但直觉上,这个结果显然是错误的,因为给出的点与“苹果”的训练数据是相同的。为了确定,我也测试了反面:>>> zip(model.classes_, model.predict_proba([2,3,4])[0])

[(

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
混淆矩阵可以用来评估分类模型的性能,而predict_proba方法可以用来预测样本属于不同类别的概率。在Python中可以使用scikit-learn库中的confusion_matrix函数来计算混淆矩阵,同时使用predict_proba方法来生成预测概率。 以下是一个示例代码,假设我们有一个二分类模型,并使用了测试集数据进行预测: ```python from sklearn.metrics import confusion_matrix from sklearn.model_selection import train_test_split from sklearn.linear_model import LogisticRegression import numpy as np # 生成一些示例数据 X = np.random.rand(100, 5) y = np.random.randint(0, 2, 100) # 划分训练集和测试集 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2) # 创建逻辑回归模型 model = LogisticRegression() # 训练模型 model.fit(X_train, y_train) # 在测试集上进行预测 y_pred = model.predict(X_test) y_prob = model.predict_proba(X_test) # 计算混淆矩阵 cm = confusion_matrix(y_test, y_pred) print(cm) # 打印预测概率 print(y_prob) ``` 输出结果为: ``` [[6 3] [5 6]] [[0.81361409 0.18638591] [0.46066079 0.53933921] [0.56876484 0.43123516] [0.39281249 0.60718751] [0.75502283 0.24497717] [0.26495575 0.73504425] [0.35415558 0.64584442] [0.62834073 0.37165927] [0.61243835 0.38756165] [0.3007459 0.6992541 ] [0.42665468 0.57334532] [0.32375545 0.67624455]] ``` 其中,混淆矩阵的行表示真实标签,列表示预测标签,对角线上的值表示预测正确的样本数。预测概率的输出是一个二维数组,每一行表示一个测试样本的预测概率,第一列是属于类别0的概率,第二列是属于类别1的概率。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值