类的数量必须大于一个；拟合 svm 时出现 1 类错误

Question

在使用 svm 时尝试拟合训练特征和标签，但说我只有一个类标签

下面是我的代码：

from sklearn.svm import SVC
import numpy as np
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import LabelEncoder

dataset=dataset.to_numpy()

features = dataset[:,:-1]
labels = dataset[:,-1]


#label enc
encode = LabelEncoder()
enclabels = encode.fit_transform(labels)

#split data
train_features, test_features, train_labels, test_labels = train_test_split(features, enclabels, test_size=0.2)
train_labels = train_labels.reshape(-1,1)

# train model
model = SVC(C=1.0, kernel="rbf")

classifier = SklearnClassifier(model=model)

#  Train  classifier
classifier.fit(train_features, train_labels)

我用了代码

print(np.unique(train_labels))

它返回：

[0 1]