ValueError:未知标签类型:'continuous',SVC Sklearn

时间:2018-02-12 03:36:23

标签: python machine-learning scikit-learn svm

我正在尝试使用sklearn名为SVC的库。

但是,当我运行程序时出现此错误:

ValueError: Unknown label type: 'continuous'

我不知道支持向量回归器是否有一个回归器库,这是我到目前为止唯一找到的库。这是我的代码:

import sklearn
from sklearn.model_selection import train_test_split
import matplotlib.pyplot as plt
import numpy as np
from sklearn.svm import SVC

X, Y = get_data(filename)

X_train, X_test, y_train, y_test = train_test_split(X, Y, random_state=33)

svc = SVC()
svc.fit(X_train, y_train)

print(svc.score(X_train, y_train))
print(svc.score(X_test, y_test))

感谢。

1 个答案:

答案 0 :(得分:3)

SVC是一个分类器,因此不支持目标中的连续值。你需要的是SVR。只需用SVR替换SVC的所有出现,你就可以了。

from sklearn.svm import SVR
svr = SVR()
svr.fit(X_train, y_train)
print(svr.score(X_train, y_train))
print(svr.score(X_test, y_test))