torch>=1.12.0
torch_geometric
scikit-learn>=1.2.2
