- 基于的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折交叉划分训练集和验证集标签。