2013-06-05 1 views
7

Je commencer avec scikit-learn ...Comment savoir quelles classes sont représentées dans le tableau de retour de predict_proba dans 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]]) 

Comment puis-je savoir quelle classe est censé être qui?

Répondre

10

Les résultats de prédiction appartiennent aux classes dans cet ordre: model.classes_

+0

Est-ce vrai pour tous les modèles? – sushmit

Questions connexes