pycharm之argparse与add_argument的用法

在学习LarsBentsen/FFTransformer: Multi-Step Spatio-Temporal Forecasting: https://authors.elsevier.com/sd/article/S0306-2619(22)01822-0 (github.com)

 时碰到了一大段的

import random
import numpy as np

def main():
    fix_seed = 2022
    random.seed(fix_seed)
    torch.manual_seed(fix_seed)
    np.random.seed(fix_seed)

    parser = argparse.ArgumentParser(description='FFTransformer, Transformer family, LSTM and MLP for Wind Forecasting')

    # basic config
    parser.add_argument('--is_training', type=int, required=False, default=1, help='status')
    parser.add_argument('--model_id', type=str, required=False, default='test', help='model id for saving')
    parser.add_argument('--model', type=str, required=False, default='FFTransformer',
                        help='model name, options: [FFTransformer, Autoformer, Informer, Transformer, LogSparse, LSTM, MLP, persistence (and same with GraphXxxx)]')
    parser.add_argument('--plot_flag', type=int, default=1, help='Whether to save loss plots or not')
    parser.add_argument('--test_dir', type=str, default='', help='Base dir to save test results')
    parser.add_argument('--verbose', type=int, default=1, help='Whether to print inter-epoch losses.')

 1、在新工程下的main.py中赋值以下代码

import math
import argparse  # 导入argparse模块

# 用来装载参数的容器
parser = argparse.ArgumentParser(description='Calculate volume of a cylinder')
# 给这个解析对象添加命令行参数
parser.add_argument('radius', type=int, help='Radius of cylinder')
parser.add_argument('height', type=int, help='Height of cylinder')
args = parser.parse_args()  # 获取所有参数


def cylinder_volume(radius, height):
    vol = (math.pi) * (radius ** 2) * (height)
    return vol


if __name__ == '__main__':
    print(cylinder_volume(args.radius, args.height))

 2、查看帮助信息

在terminal输入命令

 python .\main.py --h

 3、输入参数

参考：

python之argparse模块常见用法包含实例(超详细)_小白菜儿929的博客-CSDN博客相信很多刚开始学习深度学习算法代码的小伙伴，在一开始总会遇到关于argparse模块的代码，网上关于这方面的文章有很多但新手入门往往不是那么容易。本文从一个计算圆柱体体积的实例出发，帮助大家理解关于argparse模块的一些常见用法。点击查看官方参考文档1.介绍argparse模块是命令行选项、参数和子命令解析器。可以让人轻松编写用户友好的命令行接口。适用于代码需要频繁地修改参数的情况。先看下面一段没有使用argparse的代码import math # 为了获取πdef cylinderhttps://blog.csdn.net/RudeTomatoes/article/details/117003291python之parser.add_argument()用法——命令行选项、参数和子命令解析器_夏普通的博客-CSDN博客https://blog.csdn.net/qq_34243930/article/details/106517985