我正在尝试使用SVM拟合模型,但出现以下错误:
var questions = ["Chocolate is used in what flavour of ice cream", "Chocolate-ice-cream", ...];
此错误位于此代码的fit函数中:
ValueError: setting an array element with a
sequence.
X_train和Y_train的项目数相同。
X_train和Y_train最初是使用以下代码创建的列表:
from sklearn import svm
from numpy import array
X_tn = array(X_train)
Y_tn = array(Y_train)
modelo = svm.SVC(kernel = 'linear', C =
1.0)
X_tn = X_tn.reshape(Y_tn.shape)
modelo.fit(X_tn,Y_tn)
modelo.score(X_tn,Y_tn)
prediccion = modelo.predict(X_test)
print(prediccion)
训练集是图像,测试集是标签,对于室内图像为0,对于室外图像为1。
如何拟合模型?
谢谢