1 回答

TA貢獻(xiàn)1841條經(jīng)驗(yàn) 獲得超3個(gè)贊
sklearn 期望您的 X_train 數(shù)組是一個(gè)二維數(shù)組,例如 (n_examples, 1),而 Y_train 是一維標(biāo)簽數(shù)組,例如 (n_examples, )。
我重新格式化了您的代碼以刪除一些不必要的步驟并解決了問題:
import numpy as np
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split
from sklearn.svm import SVC
m_array = np.array([229, 230, 231, 230, 230])[:, np.newaxis]
faces = np.array(['Face 1', 'Face2', 'Face 3', 'Face 4', 'Face 5'])
X_train, X_test, y_train, y_test = train_test_split(m_array, faces, test_size = 0.20)
svclassifier = SVC(kernel='linear')
svclassifier.fit(X_train, y_train)
添加回答
舉報(bào)