基于python构建数据集5折交叉验证划分的方法

• 基于的python库为scikit-learn，通过pip install scikit-learn安装
• 相关代码如下：

from sklearn.model_selection import KFold

folds = KFold(n_splits=5, shuffle=True, random_state=seed)
for fold_i, (train_index, val_index) in enumerate(folds.split(self.image_path_list)):
    train_image_path_list = self.image_path_list[train_index]
    val_image_path_list = self.image_path_list[val_index]
    train_label_list = self.label_list[train_index]
    val_label_list = self.label_list[val_index]

通过以上代码可以5折交叉划分训练集和验证集标签。