從sklearn.svm.LinearSVC對象獲取類標籤

Question

如何從分類器對象中獲取類標籤（例如['business'，'lifestyle'，'sports'，'tech']）？分類器方法predict能夠生成標籤，所以我猜它應該存儲在分類器對象的某個位置。

（http://scikit-learn.org/stable/modules/generated/sklearn.svm.LinearSVC.html）

任何人知道如何獲取類標籤的文檔中我找不到它？

謝謝！

Answer 1

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

Answer 2

我發現它，它隱藏在對象的classes_屬性中。 閱讀源代碼後發現它。