python-3.x - 如何将数据拆分为训练和测试，以及如何在图中显示？

Question

我有 3176 条记录。我分成 80% 的训练数据和 20% 的测试数据。但是我训练了以下代码。

发生此错误。

回溯（最后一次调用）：文件“D:/Python Project/data.py”，第 179 行，在 test_results.append(tests_new[i - 2535]) IndexError: list index out of range

标签： python-3.x

Answer 1

from sklearn import model_selection
X_train, X_test, y_train, y_test = model_selection.train_test_split(train_data, target_data, test_size=0.20)

训练数据：你的特征列表

目标数据：您的预测数据