Hi Ronish, no, I used all the classes available. As J_Johnson said above, you should checkout the link on datasets and dataloaders to create a custom one which excludes the 5 class.
The way you are doing it won’t work since train_data and test_data_xy are not just lists you can directly modify.