StratifiedKFold用法類似Kfold,但是他是分層采樣,確保訓練集,測試集中各類別樣本的比例與原始數據集中相同。
例子:
import numpy as np
from sklearn.model_selection import KFold,StratifiedKFold
---------------------
作者:wqh_jingsong
來源:CSDN
原文:https://blog.csdn.net/wqh_jingsong/article/details/77896449