数据建模-SVM

Question

我目前正在执行数据建模，但遇到错误，无法找到解决方案。 因此，我希望可以从该平台获得一些帮助！ 提前致谢。

我的代码：-

from sklearn import cross_validation
from sklearn.cross_validation import cross_val_score
from sklearn.cross_validation import train_test_split
from sklearn import svm

X = np.array(observables) #X are features
y = np.array(df['diagnosis']) # y is label

X_train, y_train, X_test, y_test= cross_validation.train_test_split(X, y, test_size=0.2)

clf= svm.SVC()
clf.fit(X_train, y_train)
accuracy= clf.score(X_test, y_test)
print (accuracy)

但是我得到这个错误：

ValueError：输入形状错误（114，8）

Answer 1

似乎您混淆了train_test_split返回值的train_test_split ，请使用

X_train, X_test, y_train, y_test= cross_validation.train_test_split(X, y, test_size=0.2)

代替

X_train, y_train, X_test, y_test= cross_validation.train_test_split(X, y, test_size=0.2)