如何解決 anaconda python 中的數據重塑錯誤？

Question

我有一個示例數據，例如：姓名、AVg、GPA

這是我得到的部分數據。 我有 164 行數據

sample = pd.read_excel('./data.xlsx')
X = sample.AVG
y = sample.GPA


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

clf = DecisionTreeClassifier()
clf.fit(X_train, y_train)

y_pred = clf.predict(X_test)

當我運行上面的代碼時，我收到錯誤消息說'當 n_samples=1、test_size=0.2 和 train_size=None 時，生成的訓練集將為空。 調整任何上述參數。

所以我添加了 bleow 代碼

X =np.reshape(-1,1)
y=np.reshape(-1,1)

現在我收到錯誤消息，說“如果 n_samples=1、test_size=0.2 和 train_size=None，則生成的訓練集將為空。 調整任何上述參數。

如何刪除這些錯誤消息？ 有人可以幫幫我嗎？

Answer 1

通過 X_train=X_train.values.reshape(-1,1) y_train=y_train.values.reshape(-1,1) 解決這個問題