| 89 | |
| 90 | |
| 91 | def _parse_args(): |
| 92 | parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter, |
| 93 | conflict_handler='resolve') |
| 94 | parser.add_argument('--embedding_file', |
| 95 | help='embedding file', |
| 96 | required=True) |
| 97 | parser.add_argument('--train_label_file', |
| 98 | help='train label file', |
| 99 | required=True) |
| 100 | parser.add_argument('--test_label_file', |
| 101 | help='test label dir', |
| 102 | required=True) |
| 103 | parser.add_argument('--average', |
| 104 | default="micro", |
| 105 | help='optional: micro, macro') |
| 106 | parser.add_argument('--classifier_name', |
| 107 | default="OneVsRest", |
| 108 | help='optional: OneVsRest, SGD') |
| 109 | args = parser.parse_args() |
| 110 | return args |
| 111 | |
| 112 | |
| 113 | def main(): |