我想使用不平衡数据集评估三个模型,即 LogisticRegression
、SVM
和 Random Forest
。我决定使用一种分层的方法。
第一个选项是使用 train_test_split
并设置 stratyfy=y
但是,我使用了 StratifyKfold
方法进行了 10 次拆分。
在这种情况下,我如何使用相同的分割来评估我的三个模型?
答案 0 :(得分:1)
如果使用相同的数据集,则可以修复 random_state
的 StratifyKfold
参数。如果这样做,您将评估具有相同 10 次分割的三个模型。