python - 如何拆分训练集和测试集？

Question

我们应该在哪里使用

X_train,X_test,y_train,y_test= train_test_split(data, test_size=0.3, random_state=42)

我们应该在哪里使用

train, test= train_test_split(data, test_size=0.3, random_state=0). 

前一个返回这个:

value error: not enough values to unpack (expected 4, got 2)

Answer 1

如果要拆分具有特征 (X) 和标签 (y) 的实例，您使用的第一种形式。如果您只想拆分特征 (X)，则使用第二种形式。

X_train, X_test, y_train, y_test= train_test_split(data, y, test_size=0.3, random_state=42)

它对您不起作用的原因是您没有在 train_test_split() 函数中提供标签数据。以上应该运作良好。只需将 y 替换为您的标签/目标数据即可。