如何知道Scikit-learn中的predict_proba返回数组中表示哪些类



我从Scikit-learn开始…

>>> 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_

相关内容

  • 没有找到相关文章

最新更新