Update README.md

Blair1213 · Jan 25, 2024 · ee6059e · ee6059e
1 parent ff3f85e
commit ee6059e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -47,5 +47,10 @@ To reproduce the results of TIGER or train TIGER, you are supposed to download a
 └── data_process.py
 
 ```
+Then, you can train TIGER with the following command:
+'''
+python main.py -dataset drugbank/kegg/ogbl-biokg -extractor khop-subtree/randomWalk/probability
+'''
 
+The hyper-parameters used to train TIGER on above three datasets are shown in our paper.