File size: 356 Bytes
9c98cd3 132149b 0507e50 | 1 2 3 4 5 6 7 8 9 | # 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 |