2013-06-05 5 views
7

Sto partendo con scikit-learn ...come sapere quali classi sono rappresentate in ordine di ritorno da predict_proba in 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]]) 

Come faccio a sapere quale classe si suppone essere che?

risposta

10

La previsione risultati appartengono alle classi in questo ordine: model.classes_

+0

Questo è vero per tutti i modelli? – sushmit

Problemi correlati