运行代码工具: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)
运行结果如下: