command fix: right prompt for zero-shot classification

DeepGraphLearning · Oct 16, 2023 · db53a76 · db53a76
1 parent c90a161
commit db53a76
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -108,7 +108,7 @@ We demonstrate on zero-shot subcellular localization prediction and zero-shot re
 ```
 # Subcellular Localization Prediction
 
-python ./script/run_zero_shot.py --config ./config/zero_shot/PretrainESM/zero_shot.yaml --checkpoint $PRETRAIN_CHECKPOINT --prompt_label ./data/zero_shot_classification/subloc_name.tsv --dataset SubcellularLocalization --field "['name']"
+python ./script/run_zero_shot.py --config ./config/zero_shot/PretrainESM/zero_shot.yaml --checkpoint $PRETRAIN_CHECKPOINT --prompt_label ./data/zero_shot_classification/subloc_template.tsv --dataset SubcellularLocalization --field "['name']"
 
 # Reaction Classification