python - 拆分不平衡（二进制）数据集，因此测试集具有相同数量的假样本和正样本

Question

我正在尝试将 NN 应用于具有高级不平衡的二元分类问题；28k 假 (0.) 和 68k 真 (1.)。是否可以使用带有 train_test_split 的分层来确保测试集中 True 和 False 事件的比例相同，而不是训练集中？例如; 测试集 - 14k 错误，14k 正确。训练集 - 14k 错误，54k 正确。