如何使用 train_test_split？ 修复错误 n_samples = 0

Question

我正在尝试将我正在使用的数据拆分为训练和测试集，但是当我使用 train_test_split function 时出现 n_samples = 0 的错误。

这是我的代码：

X_train, X_test, y_train, y_test = model_selection.train_test_split(summary, labels, test_size=0.35)

摘要和标签是列表，将它们转换为 arrays 后，这是我得到的形状：

(1248,)
(1248,)

它们都有 1248 个值。 有人能告诉我为什么它不起作用吗？ 谢谢

错误信息：

With n_samples=0, test_size=0.35 and train_size=None, the resulting train set will be empty. Adjust any of the aforementioned parameters

Answer 1

对我有用，检查这是否对你有用：

from sklearn.model_selection import train_test_split
import numpy as np

# dummy examples
summary, labels = np.arange(0,1248), np.arange(0,1248)

X_train, X_test, y_train, y_test = train_test_split(summary, labels, test_size=0.35)

用字符串列表测试

summary, labels = ["hello"]*1248, ["test"]*1248