python - 预测概率

Question

尝试使用 sklearn 中的 SVC 来解决分类问题。给定一堆数据，以及告诉我某个主题是否属于某个类别的信息，我希望能够给出一个新的未知主题属于某个类别的概率。

我只有 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 

我该如何解决这个问题？

Answer 1

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

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

您的代码创建了一个带有概率估计的 SVC 并丢弃它(因为您没有将它存储在任何变量中)并使用一些以前存储在 clf 中的 SVC(没有概率)