parser.add_argument用法
时间: 2024-01-10 14:05:13 浏览: 28
parser.add_argument() 方法用于向 ArgumentParser 添加程序参数信息。通过调用该方法,可以指定 ArgumentParser 如何获取命令行字符串并将其转换为对象。这些信息在 parse_args() 调用时被存储和使用。例如,可以使用以下方式添加参数信息:
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.')
在上述示例中,'--sparse'、'--seed' 和 '--epochs' 是参数的名称,分别表示稀疏版本的 GAT、随机种子和训练的轮数。'action' 参数指定了参数的行为,'store_true' 表示如果命令行中出现了该参数,则将其值设置为 True,否则为 False。'type' 参数指定了参数的类型,'int' 表示整数类型。'default' 参数指定了参数的默认值。'help' 参数用于提供关于参数的帮助信息。
在创建 ArgumentParser 对象后,可以使用 add_argument() 方法添加参数信息。最后,使用 parse_args() 方法解析添加的参数。例如:
import argparse
parser = argparse.ArgumentParser(description='test')
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.')
args = parser.parse_args()
在上述示例中,args.sparse、args.seed 和 args.epochs 分别表示解析后的参数值。可以通过打印这些参数来查看它们的值。例如:
print(args.sparse)
print(args.seed)
print(args.epochs)
这将输出解析后的参数值。