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

【问题标题】：How to set arguements of python functions in the command prompt如何在命令提示符下设置 python 函数的参数
【发布时间】：2019-12-05 02:40:57
【问题描述】：

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

【问题讨论】：

您希望代码支持 --epoch 等？基本上，编写代码以便您可以从命令行传递这些代码？
@PySaad 是的，这样我就可以从命令行输入参数而不是 IDE 或其他的东西
使用argparse。它提供了一种将命令行参数处理为一组值的方法，然后您可以将这些值传递给您的函数。

标签： python arguments parameter-passing command-line-arguments command-prompt

【解决方案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

【讨论】：