python - ValueError : Unknown label type: 'continuous' , SVC Sklearn

Question

我正在尝试使用 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))

谢谢。

Answer 1

SVC 是一个分类器，因此不支持目标中的连续值。您需要的是SVR 。只需将所有出现的 SVC 替换为 SVR 即可。

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))