argparse
#常用库 2024-06-03

argparse 模块用于命令项选项与参数解析。

示例

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
# test.py
import argparse

parser = argparse.ArgumentParser(description='Demo of a argparse')

# 添加参数
parser.add_argument('--epochs',type=int,default=30,help="This is epochs")
parser.add_argument('--batch',type=int,default=4,help="This is batch")

args=parser.parse_args()

print(args)

epochs=args.epochs

batch=args.batch

print(f"show {epochs} {batch}")

使用命令：

1
python test2.py

输出如下：

1
2
Namespace(epochs=30, batch=4)
show 30 4

使用命令：

1
python test2.py --epochs 100 --batch 10

输出如下：

1
2
Namespace(epochs=100, batch=10)
show 100 10

使用命令：

1
python test2.py -h

输出如下：

1
2
3
4
5
6
7
8
usage: test2.py [-h] [--epochs EPOCHS] [--batch BATCH]

Demo of a argparse

optional arguments:
  -h, --help       show this help message and exit    
  --epochs EPOCHS  This is epochs
  --batch BATCH    This is batch

add_argument 的参数

  • type=int：类型

  • default=30：默认值

  • help=This is epochs：对参数的介绍信息（通过 -h 查看）

  • metavar='FILE'：帮助信息中用于表示这个参数的值应该是什么类型的。

  • action='store_true'：如果在命令行中出现了 --if_fog 这个参数，那么相应的变量（在这里是 args.if_fog）将被设置为 True。这是一种常见的用法，用于启用或禁用某些功能，而不需要显式地写出 =True 或 =False。

  • choices=['cifar10', 'cifar100']：该参数的可选值