python用argparse包定义参数

最新推荐文章于 2024-04-19 08:32:29 发布

肖蜜Fiona

最新推荐文章于 2024-04-19 08:32:29 发布

阅读量527

点赞数

分类专栏： python工具文章标签： python jupyter

原文链接：https://blog.csdn.net/weixin_33962621/article/details/94601319?spm=1001.2101.3001.6661.1&utm_medium=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1-94601319-blog-82111586.pc_relevant_3mothn_strategy_and_data_recovery&depth_1-utm_so

版权

python工具专栏收录该内容

12 篇文章 2 订阅

订阅专栏

运行代码工具：jupyter notebook

看文献代码发现在搭建神经网络时，可以用argparse包定义参数

参数一目了然，修改起来也很方便

主要有三个步骤，具体可以看代码块，

需要注意的是，如果是在jupyter notebook中运行代码，

最后一步 args = parser.parse_args() 需要修改成 args = parser.parse_args(args=[])

具体原因没查到，好像是因为jupyter会自带参数？不太清楚，有人知道可以留言告诉我吗？

#用argparse包 定义参数
import argparse
#三个步骤
#1、创建一个解析器——创建 ArgumentParser() 对象
parser = argparse.ArgumentParser(description='test')
#2、添加参数——调用 add_argument() 方法添加参数
parser.add_argument('--sparse', action='store_true', default=False, help='GAT with sparse version or not.')
parser.add_argument('--seed', type=int, default=72, help='Random seed.')
parser.add_argument('--epochs', type=int, default=10000, help='Number of epochs to train.')
#3、解析参数——使用 parse_args() 解析添加的参数
args = parser.parse_args(args=[])#在jupyter中运行括号里加入这个不然会报错，原因好像是因为jupyter自带参数

print (args)
print(args.sparse)
print(args.seed)
print(args.epochs)

运行结果如下：