如何在命令提示符中设置 python 函数的参数

Question

我希望在命令提示符中输入 python 函数的参数，如下所示， python train.py --epoch 20 --batch_size 32 --lr 0.001 等等。我应该参考什么来编写我的代码？谢谢！

Answer 1

你可以试试下面的例子-

import argparse

def some_args():
    parser = argparse.ArgumentParser(description='Some description')
    parser.add_argument('--epoch', help='Value of epoch')
    parser.add_argument('--batch-size', help='Value of batch size')
    parser.add_argument('--lr', help='Value of lr')
    return parser.parse_args()

def main():
    args = some_args()
    epoch_value = args.epoch
    batch_size_value = args.batch_size
    lr_value = args.lr
    print (epoch_value)         #prints your epoch_value which you passed
    print (batch_size_value)    #prints your batch_size_value which you passed
    print (lr_value )           #prints your lr_value which you passed


if __name__ == '__main__':
    main()

命令行参数为 -

python train.py --epoch <some_value> --batch-size <some_value> --lr <some_value>

参考

https://docs.python.org/3/howto/argparse.html

如何在命令提示符中设置 python 函数的参数

How to set arguements of python functions in the command prompt

python

arguments

command-prompt

parameter-passing

command-line-arguments