一个错误：训练代码数据集划分

xq141839 / DCSAU-Net

Elsevier-CIBM-2023: A deeper and more compact split-attention U-Net for medical image segmentation

Apache License 2.0

72 stars 16 forks source link

Closed xueyue-hd closed 1 year ago

xueyue-hd commented 1 year ago

在使用df = df[df.category == 'train']切分DataFrame数据后，默认的索引已经出现缺失，应该更新索引（df.reset_index(drop=True, inplace=True) # 重置DataFrame的索引），否则无法继续进行K折划分

xq141839 commented 1 year ago

好的，感谢你的指正，我已经更新代码了