如何知道在 Scikit-learn 中的 predict_proba 的返回数组中表示哪些类 [英] How to know what classes are represented in return array from predict_proba in Scikit-learn

问题描述

我从 Scikit-learn 开始...

<预><代码>>>>导入sklearn>>>sklearn.__version__'0.13.1'>>>从 sklearn 导入 svm>>>模型 = svm.SVC(概率=真)>>>X = [[1,2,3], [2,3,4]] # 特征向量>>>Y = ['apple', 'orange'] # 类>>>模型拟合(X，Y)>>>model.predict_proba([1,2,3])数组([[ 0.39097541, 0.60902459]])

我怎么知道哪个类应该是哪个?

解决方案

预测结果属于类的顺序如下:model.classes_

I'm starting off with 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]])

How do I know which class is supposed to be which?

解决方案

The prediction results belong to the classes in this order: model.classes_

这篇关于如何知道在 Scikit-learn 中的 predict_proba 的返回数组中表示哪些类的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！