15

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?

Rohan Nadagouda
  • 462
  • 7
  • 18
Alex
  • 18,332
  • 10
  • 49
  • 53

1 Answers1

16

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

Bilal Dadanlar
  • 820
  • 7
  • 14