分层和模型评估

Stratify and model evaluation

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

如果使用同一个数据集,可以修改StratifyKfoldrandom_state参数。如果这样做,您将评估具有相同 10 个拆分的三个模型。