sfqcyy
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

bert script bugfix

This commit is contained in:
yoonlee666 2020-08-31 15:16:40 +08:00
parent 577beb21ec
commit ff4e02a09d
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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