train_test_split用于将数据集划分为训练集和测试集。通过将数据集划分为训练集和测试集，可以在训练模型时使用训练集进行模型训练，然后使用测试集评估模型的性能。这样可以有效地评估模型在新数据上的泛化能力，避免过拟合。train_test_split还可以用于交叉验证等模型评估方法。