6

我们如何从分类器对象中获取类别标签(例如,['business','lifestyle','sports','tech'])?分类器方法predict能够产生标签,所以我猜它应该存储在分类器对象内的某个地方。

我在文档中找不到它(http://scikit-learn.org/stable/modules/generated/sklearn.svm.LinearSVC.html

任何人都知道如何获得类标签?

谢谢!

4

2 回答 2

12

有一个classes_领域。

>>> from sklearn import svm
>>> clt = svm.SVC()
>>> clt.fit( [[1],[2],[3]], ["a","b","a"] )
SVC(C=1.0, cache_size=200, class_weight=None, coef0=0.0, degree=3, gamma=0.0,
  kernel='rbf', max_iter=-1, probability=False, shrinking=True, tol=0.001,
  verbose=False)
>>> clt.classes_
array(['a', 'b'], 
      dtype='|S2')
于 2013-08-22T08:56:11.883 回答
1

我找到了,它隐藏在classes_对象的属性中。看了源码才发现。

于 2013-08-22T08:55:53.440 回答