机器学习样例提交train_test_split

这里数据集划分训练完后得到正常预测结果

这里想在我真正的测试集上去预测可是却会报错

求解决怎么才能将预测结果存入