一. 简介
argparse是python用于解析命令行参数和选项的标准模块,用于代替已经过时的optparse模块。argparse模块的作用是用于解析命令行参数。我们很多时候,需要用到解析命令行参数的程序,目的是在终端窗口(ubuntu是终端窗口,windows是命令行窗口)输入训练的参数和选项。
一. 步骤
- 导入模块:import argparse
- 创建解析对象:parser = argparse.ArgumentParser()
- 添加参数:parser.add_argument()
- 解析:parser.parse_args()
二. 例子
import argparse
parser = argparse.ArgumentParser(description='BERTRL model')
parser.add_argument("--dataset", "-d", type=str, help="Dataset string")
parser.add_argument("--hop", type=int, default=3, help="max reasoning path length")
parser.add_argument('--set_type', '-st', type=str, default='train', help='set type of train/valid/test')
params = parser.parse_args()
注:使用“python python文件名 -h”命令可以打印所有的参数