KFold 和ShuffleSplit的区别

KFold 

sklearn.model_selection.KFold(n_splits=’warn’, shuffle=False, random_state=None)

 

ShuffleSplit

sklearn.model_selection.ShuffleSplit(n_splits=10, test_size=’default’, train_size=None, random_state=None)

 

KFold把数据集分成n份，只取一份作为测试集，其余作为训练集，

ShuffleSplit把数据集分成n份，取test_size份作为测试集，train_size份作为训练集。

ShuffleSplit比KFold更加灵活。