parser.add_argument()用来加载数据集
时间: 2023-07-24 20:20:25 浏览: 91
代码细节处理学习(1)——如何使用argparse-python(3)——add_argument()用法
`parser.add_argument()` 是用来添加命令行参数的方法,它的作用是告诉argparse解析器需要解析哪些参数以及这些参数的具体信息。在这段代码中,`parser.add_argument('--dataset', type=str, default='cora', help='Dataset string.')` 就是添加了一个名为"--dataset"的命令行参数,它的类型为字符串,缺省值为'cora',并且添加了一个帮助信息"Dataset string.",用于在用户使用"--help"参数时显示。
当程序运行时,argparse解析器会根据这些参数信息来解析命令行参数,并将解析结果存储到args变量中。在这个例子中,我们可以使用命令行参数"--dataset citeseer"来指定要加载Citeseer数据集,例如:
```
python main.py --dataset citeseer
```
这里的"--dataset citeseer"就是命令行参数,它会被argparse解析器识别为一个名为"--dataset"的参数,并将其值设置为"citeseer",程序会加载Citeseer数据集。
阅读全文