self.train = attack_data[randIdx[:N_train]] self.train_labels = attack_labels[randIdx[:N_train]] trained data is generated in the data_loader.py line 38 and 39. But why only attacked data used but not normal data.