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

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). 

前一个返回这个：

值错误：没有足够的值来解包（预期为 4，得到 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为您的标签/目标数据即可。