Взяв слегка измененную версию примера, приведенного в документации :
import numpy as np
X = np.array([[-1, -1], [-2, -1], [1, 1], [2, 1]])
y = np.array([-1, -1, 1, 1])
from sklearn.svm import SVC
clf = SVC()
clf.fit(X, y)
print(f'Number of support vectors in each class: {clf.n_support_}')
print(f'Classes: {clf.classes_}')
, вы можете получить доступ к классам классификатора, вызвав .classes_
.
Приведенный выше код распечатывает:
Количество векторов поддержки в каждом классе: [2 2]
Классы: [-1 1]
что означает, что 2 принадлежат классу -1
, 2 принадлежат классу 1
.