本文介绍了如何知道在 Scikit-learn 中的 predict_proba 的返回数组中表示哪些类的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
限时送ChatGPT账号..我从 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屋!
更多推荐
[db:关键词]
发布评论