从sklearn.svm.LinearSVC对象获取类标签

时间:2013-08-22 08:47:13

标签: python machine-learning

我们如何获得课程标签(例如,[' business',' lifestyle',' sports' tech' tech'] )来自分类器对象?分类器方法predict能够生成标签,所以我猜它应该存储在分类器对象中的某个位置。

我无法在文档(http://scikit-learn.org/stable/modules/generated/sklearn.svm.LinearSVC.html

中找到它

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

谢谢!

2 个答案:

答案 0 :(得分:8)

有一个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')

答案 1 :(得分:1)

我发现它,它隐藏在对象的classes_属性中。 在阅读源代码后找到它。