检查各数据集的长度

为确认 train_test_split() 按预期运行，请检查 X_train 和 X_test 的长度，看看每个数据集中有多少条记录。您可以使用 len() 等函数，或 .shape 等属性来查看。