分层和模型评估

Question

我希望使用不平衡的数据集评估三个模型，即LogisticRegression，SVM和Random Forest。我决定使用一种条纹化的方法。第一个选项是使用train_test_split并设置stratyfy=y，无论我如何使用10个拆分的StratifyKfold方法。在这种情况下，如何使用相同的拆分来评估我的三个模型？

Answer 1

如果使用相同的数据集，则可以修复StratifyKfold的random_state参数。如果这样做，您将使用相同的10个拆分来评估这三个模型。