skf = StratifiedKFold(n_splits=5, shuffle = True, random_state=321)
时间: 2023-06-06 11:09:02 浏览: 82
一份数据随机分成n等份
5星 · 资源好评率100%
这是一个机器学习中的交叉验证方法,将数据集分成几个相等的部分,然后将每个部分轮流作为测试集和训练集。 stratify 参数用于确保每个部分中的类别比例与整个数据集中的类别比例相同。 n_splits 参数指定划分的部分数量, shuffle 参数表示是否打乱数据集顺序, random_state 参数表示随机数种子,用于确保结果的可重复性。
阅读全文