train_test_split的用法stratify

参数的作用是什么？能否举个例子说明？谢谢。

train_test_split函数是用来将数据集划分为训练集和测试集的函数。其中，stratify参数的作用是使划分后的训练集和测试集中各类别数据的比例与整个数据集中各类别数据的比例尽量一致。

举个例子来说，对于一个二分类问题，数据集中有200个数据样本，其中100个样本属于类别A，100个样本属于类别B。如果不使用stratify参数进行划分，则划分后的训练集和测试集可能会出现某个类别的数据比例明显偏高或偏低的情况。

而如果使用stratify参数并将原始数据集的标签作为参数传入，则可以确保训练集和测试集中各类别数据的比例与整个数据集中各类别数据的比例尽量一致，从而更加准确地评估模型性能。