我们在写一个成熟的Python项目时候需要传入若干指定的参数。而不是写死在程序里这个时候就要用到argparse模块。argparse 是 Python 内置的一个用于命令项选项与参数解析的模块通过在程序中定义好我们需要的参数argparse 将会从sys.argv 中解析出这些参数并自动生成帮助和使用信息。

一、使用步骤

1. 创建 ArgumentParser() 对象
2. 调用 add_argument() 方法添加参数
3. 使用 parse_args() 解析添加的参数

二、举例

# 初始化
parser = argparse.ArgumentParser()
# 定义参数类型
parser.add_argument('-p', type=str, default='train', help='train | test')
parser.add_argument('--dataset', type=str, default='book', help='book | taobao')
parser.add_argument('--random_seed', type=int, default=19)
parser.add_argument('--embedding_dim', type=int, default=64)
parser.add_argument('--hidden_size', type=int, default=64)
parser.add_argument('--num_interest', type=int, default=4)
parser.add_argument('--learning_rate', type=float, default=0.001, help='')
parser.add_argument('--max_iter', type=int, default=1000, help='(k)')
parser.add_argument('--patience', type=int, default=50)
parser.add_argument('--coef', default=None)
parser.add_argument('--topN', type=int, default=50)
# 调用参数
args = parser.parse_args()
SEED = args.random_seed

运行结果如下