search

muhanzhang / pytorch_DGCNN

PyTorch implementation of DGCNN
MIT License
372 stars 122 forks source link

Suffling dataset #41

Closed andreitam11 closed 3 years ago

andreitam11 commented 3 years ago

Hello. Is there a way to shuffle the dataset? I have the graphs in the .txt, but unfortunately, the similarly labeled graphs are grouped together. Is there a way to shuffle them before running?

muhanzhang commented 3 years ago

Yes. You can randomly shuffle the train_graphs after loading them from txt by 

random.shuffle(train_graphs)