无法为SVM拟合模型

时间:2018-10-11 21:38:54

标签: python numpy scikit-learn svm

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

如何拟合模型?

谢谢

0 个答案:

没有答案