diff --git a/model_zoo/official/nlp/bert/run_classifier.py b/model_zoo/official/nlp/bert/run_classifier.py index 947ebc3438..dbc461773a 100644 --- a/model_zoo/official/nlp/bert/run_classifier.py +++ b/model_zoo/official/nlp/bert/run_classifier.py @@ -136,9 +136,10 @@ def run_classifier(): parser = argparse.ArgumentParser(description="run classifier") parser.add_argument("--device_target", type=str, default="Ascend", choices=["Ascend", "GPU"], help="Device type, default is Ascend") - parser.add_argument("--assessment_method", type=str, default="accuracy", - choices=["Mcc", "Spearman_correlation", "Accuracy"], - help="assessment_method including [MCC, Spearman_correlation, Accuracy], default is accuracy") + parser.add_argument("--assessment_method", type=str, default="Accuracy", + choices=["Mcc", "Spearman_correlation", "Accuracy", "F1"], + help="assessment_method including [Mcc, Spearman_correlation, Accuracy, F1],\ + default is Accuracy") parser.add_argument("--do_train", type=str, default="false", choices=["true", "false"], help="Enable train, default is false") parser.add_argument("--do_eval", type=str, default="false", choices=["true", "false"], diff --git a/model_zoo/official/nlp/bert/run_ner.py b/model_zoo/official/nlp/bert/run_ner.py index 1730de8542..069201dfa3 100644 --- a/model_zoo/official/nlp/bert/run_ner.py +++ b/model_zoo/official/nlp/bert/run_ner.py @@ -145,7 +145,7 @@ def run_ner(): parser = argparse.ArgumentParser(description="run classifier") parser.add_argument("--device_target", type=str, default="Ascend", choices=["Ascend", "GPU"], help="Device type, default is Ascend") - parser.add_argument("--assessment_method", type=str, default="accuracy", choices=["f1", "clue_benchmark"], + parser.add_argument("--assessment_method", type=str, default="F1", choices=["F1", "clue_benchmark"], help="assessment_method include: [F1, clue_benchmark], default is F1") parser.add_argument("--do_train", type=str, default="false", choices=["true", "false"], help="Eable train, default is false")