python - 预测概率

Question

尝试使用SVCfromsklearn来做分类问题。给定一堆数据，以及告诉我某个科目是否属于某个班级的信息，我希望能够给出一个新的未知科目属于某个班级的概率。

我只有 2 个类，所以问题是二进制的。这是我的代码和一些错误

from sklearn.svm import SVC
clf=SVC()

clf=clf.fit(X,Y)


SVC(probability=True)
print clf.predict_proba(W) #Error is here

但它返回以下错误：

NotImplementedError: probability estimates must be enabled to use this method

我怎样才能解决这个问题？

score 5 · Accepted Answer

您必须使用构建 SVC 对象probability=True

from sklearn.svm import SVC
clf=SVC(probability=True)
clf.fit(X,Y)
print clf.predict_proba(W) #No error

您的代码创建一个带有概率估计的 SVC 并将其丢弃（因为您没有将其存储在任何变量中）并使用存储在 clf 中的一些先前的 SVC（没有概率）

score -1 · Accepted Answer

始终在拟合之前设置参数。

from sklearn.svm import SVC
clf=SVC(probability=True)
clf=clf.fit(X,Y)
print clf.predict_proba(W)

2 回答 2