from sklearn.model_selection import train_test_split

from sklearn.model_selection import train_test_split是一个用于将数据集分成训练集和测试集的函数。这个函数可以将数据集按照一定比例分成训练集和测试集，以便我们可以使用训练集来训练模型，使用测试集来评估模型的性能。其中，train_data和train_target是要被分割的数据集和对应的标签，test_size是测试集所占的比例，random_state是随机数种子，stratify是为了保证训练集和测试集中各类别样本的比例相同。这个函数是机器学习中非常常用的一个函数，可以帮助我们更好地评估模型的性能。