summaryrefslogtreecommitdiff
path: root/requirements.txt
diff options
context:
space:
mode:
Diffstat (limited to 'requirements.txt')
-rw-r--r--requirements.txt8
1 files changed, 8 insertions, 0 deletions
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..1190cff
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,8 @@
+torch>=2.0
+torch_geometric>=2.4
+torch_sparse
+torch_scatter
+numpy
+scipy
+scikit-learn
+matplotlib