Python ArgumentParse的subparser用法說明
更新時間：2020年04月20日 10:43:08 作者：咸魚醬
這篇文章主要介紹了Python ArgumentParse的subparser用法說明，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
在寫一些很小的機(jī)器學(xué)習(xí)項(xiàng)目的時候,我們往往希望training, testing和inference能共用一個入口main,但是不同的功能使用不同的input參數(shù).當(dāng)然如果三個功能對應(yīng)三個.py腳本問題也不大,但是畢竟覺得不太優(yōu)雅.這個時候就需要考慮如何讓代碼更加簡單有條理.
主要是最近在看parser有關(guān)的東西,所以看到了一個項(xiàng)目,里面的使用subparser的地方是值得借鑒的,下面附上代碼和部分自己的一些見解
def main():
parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers()
hparams = make_hparams()
# 這個函數(shù)是直接寫了一些超參數(shù),講真我不太喜歡這個操作,個人還是比較傾向用一個額外的config文件來存儲
# 這些超參,這樣輸入的只要是config文件的路徑即可;主要是這么做可以看到自己每一步的參數(shù)是怎么設(shè)置的
# 便于后期出現(xiàn)了問題來排錯
subparser = subparsers.add_parser("train")
# add subparser here
subparser.set_defaults(callback=lambda args: run_train(args, hparams))
# 加上callback選項(xiàng),run_train是前期定義的一個函數(shù),這條和后面的args.callback(args)對應(yīng)
hparams.populate_arguments(subparser)
# 這里就是作者自己定義的一個函數(shù),本質(zhì)其實(shí)還是一系列的add_argument
subparser.add_argument("--numpy-seed", type=int)
subparser.add_argument("--model-path-base", required=True)
subparser.add_argument("--evalb-dir", default="EVALB/")
subparser.add_argument("--train-path", default="data/02-21.10way.clean")
subparser.add_argument("--dev-path", default="data/22.auto.clean")
subparser.add_argument("--batch-size", type=int, default=250)
subparser.add_argument("--subbatch-max-tokens", type=int, default=2000)
subparser.add_argument("--eval-batch-size", type=int, default=100)
subparser.add_argument("--epochs", type=int)
subparser.add_argument("--checks-per-epoch", type=int, default=4)
subparser.add_argument("--print-vocabs", action="store_true")
subparser = subparsers.add_parser("test")
subparser.set_defaults(callback=run_test)
subparser.add_argument("--model-path-base", required=True)
subparser.add_argument("--evalb-dir", default="EVALB/")
subparser.add_argument("--test-path", default="data/23.auto.clean")
subparser.add_argument("--test-path-raw", type=str)
subparser.add_argument("--eval-batch-size", type=int, default=100)
subparser = subparsers.add_parser("ensemble")
subparser.set_defaults(callback=run_ensemble)
subparser.add_argument("--model-path-base", nargs='+', required=True)
subparser.add_argument("--evalb-dir", default="EVALB/")
subparser.add_argument("--test-path", default="data/22.auto.clean")
subparser.add_argument("--eval-batch-size", type=int, default=100)
subparser = subparsers.add_parser("parse")
subparser.set_defaults(callback=run_parse)
subparser.add_argument("--model-path-base", required=True)
subparser.add_argument("--input-path", type=str, required=True)
subparser.add_argument("--output-path", type=str, default="-")
subparser.add_argument("--eval-batch-size", type=int, default=100)
subparser = subparsers.add_parser("viz")
subparser.set_defaults(callback=run_viz)
subparser.add_argument("--model-path-base", required=True)
subparser.add_argument("--evalb-dir", default="EVALB/")
subparser.add_argument("--viz-path", default="data/22.auto.clean")
subparser.add_argument("--eval-batch-size", type=int, default=100)
args = parser.parse_args()
args.callback(args)
補(bǔ)充知識：python 學(xué)習(xí)筆記--argparse模塊以及parse_known_args(）函數(shù)
代碼test.py：
import argparse
import sys
parse=argparse.ArgumentParser()
parse.add_argument("--learning_rate",type=float,default=0.01,help="initial learining rate")
parse.add_argument("--max_steps",type=int,default=2000,help="max")
parse.add_argument("--hidden1",type=int,default=100,help="hidden1")
flags,unparsed=parse.parse_known_args(sys.argv[1:])
print flags.learning_rate
print flags.max_steps
print flags.hidden1
print unparsed
運(yùn)行
python test.py --learning_rate 20 --max_steps 10 --hidden1 100 --arg_int 2
其效果等同于
python test.py --learning_rate=20 --max_steps=10 --hidden1=100 --arg_int=2
輸出：
20.0
10
100['--arg_int', '2']
flags為namespace空間，結(jié)果是Namespace(hidden1=100, learning_rate=20.0, max_steps=10)，包含程序定義了的命令行參數(shù)，而unparsed為程序沒有定義的命令行參數(shù)。
以上這篇Python ArgumentParse的subparser用法說明就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
本文主要介紹了Python中re.findall()用法詳解，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-07-07
Django框架實(shí)現(xiàn)的分頁demo示例
這篇文章主要介紹了Django框架實(shí)現(xiàn)的分頁demo,結(jié)合實(shí)例形式分析了Django框架分頁的步驟、原理、相關(guān)操作技巧與注意事項(xiàng),需要的朋友可以參考下2019-05-05
Python內(nèi)置函數(shù)hex()的實(shí)現(xiàn)示例
這篇文章主要介紹了Python內(nèi)置函數(shù)hex()的實(shí)現(xiàn)示例，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-08-08
這篇文章主要介紹了Python?中OS?module的使用詳解,本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2023-03-03
Python編程語言的作用非常強(qiáng)大，而且其應(yīng)用方便的特點(diǎn)也對開發(fā)人員起到了非常大的作用。在這里我們就可以先從Python流程控制關(guān)鍵字的相關(guān)概念開始了解，從而初步掌握這一語言的特點(diǎn)2016-03-03
Python 利用高德地圖api實(shí)現(xiàn)經(jīng)緯度與地址的批量轉(zhuǎn)換
這篇文章主要介紹了Python 利用高德地圖api實(shí)現(xiàn)經(jīng)緯度與地址的批量轉(zhuǎn)換，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-08-08
這篇文章主要介紹了Python 使用matplotlib模塊模擬擲骰子，本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，具有一定的參考借鑒價值,需要的朋友可以參考下2019-08-08
這篇文章主要介紹了Python的列表和列表算法，具有一定參考價值，需要的朋友可以了解下，希望能給你帶來幫助
2021-08-08 
舉例講解Linux系統(tǒng)下Python調(diào)用系統(tǒng)Shell的方法
這篇文章主要介紹了舉例講解Linux系統(tǒng)下Python調(diào)用系統(tǒng)Shell的方法,包括用Python和shell讀取文件某一行的實(shí)例,需要的朋友可以參考下
2015-11-11