# 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