分层和模型评估

Question

我想使用不平衡的数据集评估三个模型，即LogisticRegression、SVM 和Random Forest。我决定使用严格的方法。 第一个选项是使用train_test_split 并设置stratyfy=y 但是，我使用了 StratifyKfold 方法进行了 10 次拆分。 在这种情况下，我如何使用相同的拆分来评估我的三个模型？

Answer 1

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