torch==1.0.1.post2
ordered_set==3.1
numpy==1.16.2
torch_scatter==1.1.2
scikit_learn==0.21.1