Browse Source

!5568 bert script bugfix

Merge pull request !5568 from yoonlee666/master
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
042ac51f05
2 changed files with 5 additions and 4 deletions
  1. +4
    -3
      model_zoo/official/nlp/bert/run_classifier.py
  2. +1
    -1
      model_zoo/official/nlp/bert/run_ner.py

+ 4
- 3
model_zoo/official/nlp/bert/run_classifier.py View File

@@ -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"],


+ 1
- 1
model_zoo/official/nlp/bert/run_ner.py View File

@@ -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")


Loading…
Cancel
Save