ArgConfig class for reciving parameters
| 52 | |
| 53 | |
| 54 | class ArgConfig(configparser.ConfigParser): |
| 55 | """ArgConfig class for reciving parameters""" |
| 56 | def __init__(self, args=None): |
| 57 | super(ArgConfig, self).__init__() |
| 58 | |
| 59 | parser = argparse.ArgumentParser(description="BaiDu's Denpendency Parser.") |
| 60 | model_g = ArgumentGroup(parser, "model", "model configuration and paths.") |
| 61 | model_g.add_arg( |
| 62 | "--mode", |
| 63 | default="train", |
| 64 | choices=["train", "evaluate", "predict", "predict_q"], |
| 65 | help="Select task mode", |
| 66 | ) |
| 67 | model_g.add_arg("--config_path", "-c", default="config.ini", help="path to config file") |
| 68 | model_g.add_arg( |
| 69 | "--model_files", |
| 70 | default="model_files/baidu", |
| 71 | help="Directory path to save model and ", |
| 72 | ) |
| 73 | |
| 74 | data_g = ArgumentGroup(parser, "data", "Data paths, vocab paths and data processing options") |
| 75 | data_g.add_arg("--train_data_path", help="path to training data.") |
| 76 | data_g.add_arg("--valid_data_path", help="path to valid data.") |
| 77 | data_g.add_arg("--test_data_path", help="path to testing data.") |
| 78 | data_g.add_arg("--infer_data_path", help="path to dataset") |
| 79 | data_g.add_arg("--batch_size", default=1000, type=int, help="batch size") |
| 80 | |
| 81 | log_g = ArgumentGroup(parser, "logging", "logging related") |
| 82 | log_g.add_arg("--log_path", default="./log/log", help="log path") |
| 83 | log_g.add_arg( |
| 84 | "--log_level", |
| 85 | default="INFO", |
| 86 | choices=["DEBUG", "INFO", "WARNING", "ERROR", "FATAL"], |
| 87 | help="log level", |
| 88 | ) |
| 89 | log_g.add_arg( |
| 90 | "--infer_result_path", |
| 91 | default="infer_result", |
| 92 | help="Directory path to infer result.", |
| 93 | ) |
| 94 | |
| 95 | run_type_g = ArgumentGroup(parser, "run_type", "running type options.") |
| 96 | run_type_g.add_arg( |
| 97 | "--use_cuda", |
| 98 | "-gpu", |
| 99 | action="store_true", |
| 100 | help="If set, use GPU for training.", |
| 101 | ) |
| 102 | run_type_g.add_arg( |
| 103 | "--preprocess", |
| 104 | "-p", |
| 105 | action="store_true", |
| 106 | help="whether to preprocess the data first", |
| 107 | ) |
| 108 | run_type_g.add_arg( |
| 109 | "--use_data_parallel", |
| 110 | action="store_true", |
| 111 | help="The flag indicating whether to use data parallel mode to train the model.", |