train_test_split stratify

train_test_split是一种用于将数据集分成训练集和测试集的方法，可以帮助我们评估模型的性能。stratify是train_test_split中的一个参数，用于按照指定的标签分层抽样，确保训练集和测试集中的标签比例相同。这样可以避免出现训练集和测试集中标签分布不均匀的情况，从而提高模型的泛化能力。