python SVM报错 ValueError: The number of classes has to be greater than one; got 1 class

在使用python自带的SVM库训练生成二分类器的时候报了这个错:

ValueError: The number of classes has to be greater than one; got 1 class

在网上一番查找后发现,svm的y需要至少两种值,所以这个报错意味着,我的y只有一个值

本例使用的y中标签只有0和1两种值

让我们来看看这段代码:

x_train, x_test, y_train, y_test = train_test_split(x, y, random_state=1, train_size=0.6)
clf = svm.SVC(C=0.1, kernel='linear',decision_function_shape='ovo')

y_train是从原始的y中分割出来的训练集,这意味着y_train极有可能被分成只有0值。输出y_train查看,果然只有0值

解决方法:在原数据表中把原本在最后几行的1值调到前面,使y_train有可能包含1,这样就有两类数据标签了

你可能感兴趣的:(Debugging记录,python,svm)