| # NOTE: TopoEncoder.py contains a case HGT-based model that shows can be trained on KG-dataset. | |
| # You can refer to it for potential model refinement for REDDA | |
| python prediction.py \ | |
| --input_csv drug_disease_indication_drug_5.csv \ | |
| --target_disease DISEASE_10059 \ | |
| --output_folder results/run2 \ | |
| --device_id 0 \ | |
| --dataset KGdataset_tiny |