Unverified Commit 21363fea authored by NingMa's avatar NingMa Committed by GitHub

Update README.md

parent 4baa27f8
...@@ -2,13 +2,17 @@ ...@@ -2,13 +2,17 @@
For more details, please refer to our paper "Adaptive-Step Graph Meta-Learner for Few-Shot Graph Classification" For more details, please refer to our paper "Adaptive-Step Graph Meta-Learner for Few-Shot Graph Classification"
## Environments ## Environments
python 3.6 - python 3.6
pytorch 1.3.0 - pytorch 1.3.0
torch-cluster 1.4.5 - torch-cluster 1.4.5
torch-geometric 1.3.2 - torch-geometric 1.3.2
torch-scatter 1.4.0 - torch-scatter 1.4.0
torch-sparse 0.4.3 - torch-sparse 0.4.3
## Dataset ## Dataset
We provide For origin TRIANGLES dataset, you can download from here. In experiments, we use TRIANGLES with the partition rules of Jatin Chauhan.
## Training and Test ## Training and Test
To train the AS-MAML framework with GraghSAGE and SAGPool, please run:
python main.py
To test the trained model, please run the following:
python test.py
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment