labels包含0,1数据,但是X_train, X_test, y_train, y_test = train_test_split(sequences, labels, test_size=0.2, random_state=30)中y_train不包含1的原因
时间: 2023-10-23 15:38:59 浏览: 57
kNN.zip_KNN 分类 python_knn_knn python实现_knn分类python_数字数据集
可能是因为在划分数据集时,使用了参数stratify=labels,这样可以确保训练集和测试集中的0和1的比例与原始数据集中一致。如果数据集中1的数量比较少,不使用stratify可能会导致训练集中1的数量过少,从而影响模型的效果。所以通过stratify参数来确保训练集和测试集中的1和0的比例一致,有助于提高模型的准确度。
阅读全文