CAMMA-public / rendezvous

A transformer-inspired neural network for surgical action triplet recognition from laparoscopic videos.
Other
24 stars 8 forks source link

Fix deleting data and data leakage #6

Closed felixholm closed 2 years ago

felixholm commented 2 years ago

Calling train_videos = train_videos[:-5] val_videos = train_videos[-5:] in this order first deletes the last 5 videos from train videos and then assigns the new last 5 to validation, resulting in unused data and data leakage. This can be fixed by swapping the order of these calls val_videos = train_videos[-5:] train_videos = train_videos[:-5] such that the 5 videos are first assigned as val and then removed from train.