将模型改为交叉验证后,数据加载时提示KeyError:193,
猜测是调用__getitem__时没有找到索引为193的样本,几次排查后发现是使用KFold划分交叉验证的训练集和验证集后,没有重置索引,这里的193报错应该是索引为193的样本不在训练集的DataLoader中,而是在测试集中
解决方法,在划分训练集和验证集的代码后加两行重置索引的代码即可
train_df_fold = train_df.iloc[train_ids]
test_df_fold = train_df.iloc[test_ids]
# 重置索引
train_df_fold = train_df_fold.reset_index(drop=True)
test_df_fold = test_df_fold.reset_index(drop=True)