一、argparse
- python用于解析命令行参数和选项的标准模块,后面可以跟着不同的参数选项以实现不同功能
- 使用 argparse 配置命令行参数 步骤:
1、创建 ArgumentParser() 对象
2、调用 add_argument() 方法添加参数
3、使用 parse_args() 解析添加的参数
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument("--input", type=str, default="./add_hot_app_describe.csv", help="input file")
parser.add_argument("--model_dir", type=str, required=True, help="model dir")
parser.add_argument("--output", type=str, default="./predict/result.txt", help="output file")
return parser.parse_args()