Python中argparse用法

# encoding: utf-8
import argparse

parser = argparse.ArgumentParser()
# 字符
parser.add_argument('-en', '--env-name', default='some_string')
# 整數
parser.add_argument('-e', '--epochs', type=int, default=200)
# 浮點數
parser.add_argument('-t', '--dt', type=float, default=0.25)
# Boolean
parser.add_argument('-i', '--init', default=False, action='store_true')
# 整數列表, 使用時直接空格分開就行,不需中括號和逗號
parser.add_argument('-hs', '--hidden-sizes', type=int, nargs='+', default=[64,32])
# 浮點數列表
parser.add_argument('-qls', '--q-ls', type=float, nargs='+', default=[3.0,0.0,-3.0])

args = parser.parse_args()

NUM_EPOCHS = args.epochs
hidden_sizes = args.hidden_sizes
.....
.....

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章