Python之argparse模块的使用

最新推荐文章于 2025-04-24 12:30:08 发布

堇禤

最新推荐文章于 2025-04-24 12:30:08 发布

阅读量287

点赞数

CC 4.0 BY-SA版权

分类专栏： Pytorch学习笔记文章标签： python 开发语言

本文链接：https://blog.youkuaiyun.com/CRW__DREAM/article/details/128691918

Pytorch学习笔记专栏收录该内容

4 篇文章

订阅专栏

argparse是Python内置的命令行选项和参数解析模块，用于在项目中动态传入参数，而非硬编码。通过ArgumentParser对象，我们可以添加不同类型的参数，如-p，--dataset等，并设置默认值和帮助信息。解析参数后，可以通过parse_args()获取并使用这些参数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

一、使用步骤

创建 ArgumentParser() 对象
调用 add_argument() 方法添加参数
使用 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