Requirements:
torch==1.13.1
torch-geometric==2.3.1
torch-sparse==0.6.17
torch-scatter==2.1.1
pandas==1.4.2
matplotlib==3.5.1
scikit-learn==1.0.2
pickle5