python：argparse模块

1 简介

argparse 模块是 Python 内置的用于命令项选项与参数解析的模块，argparse 模块可以让人轻松编写用户友好的命令行接口，能够帮助程序员为模型定义参数。

2 步骤

导入argparse包 ——import argparse

创建一个命令行解析器对象 ——创建 ArgumentParser() 对象

给解析器添加命令行参数 ——调用add_argument() 方法添加参数

解析命令行的参数 ——使用 parse_args() 解析添加的参数

3 简单例子

# 导入库
import argparse
 
# 1. 定义命令行解析器对象
parser = argparse.ArgumentParser()
 
# 2. 添加命令行参数
parser.add_argument('--epochs', type=int, required=True)        # 必要参数
parser.add_argument('--batch', type=int, default=4)             # 默认值

 
# 3. 从命令行中结构化解析参数
args = parser.parse_args()
epochs = args.epochs
batch = args.batch

4 执行

python [文件名] --epochs EPOCHS [--batch BATCH]

python my_file.py -epochs 100 --batch 128