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