在寫一些很小的機(jī)器學(xué)習(xí)項(xiàng)目的時(shí)候,我們往往希望training, testing和inference能共用一個(gè)入口main,但是不同的功能使用不同的input參數(shù).當(dāng)然如果三個(gè)功能對(duì)應(yīng)三個(gè).py腳本問題也不大,但是畢竟覺得不太優(yōu)雅.這個(gè)時(shí)候就需要考慮如何讓代碼更加簡單有條理.

主要是最近在看parser有關(guān)的東西,所以看到了一個(gè)項(xiàng)目,里面的使用subparser的地方是值得借鑒的,下面附上代碼和部分自己的一些見解

def main():

parser = argparse.ArgumentParser()

subparsers = parser.add_subparsers()

hparams = make_hparams()

# 這個(gè)函數(shù)是直接寫了一些超參數(shù),講真我不太喜歡這個(gè)操作,個(gè)人還是比較傾向用一個(gè)額外的config文件來存儲(chǔ)

# 這些超參,這樣輸入的只要是config文件的路徑即可;主要是這么做可以看到自己每一步的參數(shù)是怎么設(shè)置的

# 便于后期出現(xiàn)了問題來排錯(cuò)

subparser = subparsers.add_parser("train")

# add subparser here

subparser.set_defaults(callback=lambda args: run_train(args, hparams))

# 加上callback選項(xiàng),run_train是前期定義的一個(gè)函數(shù),這條和后面的args.callback(args)對(duì)應(yīng)

hparams.populate_arguments(subparser)

# 這里就是作者自己定義的一個(gè)函數(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ǔ)充知識(shí)：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)容了，希望能給大家一個(gè)參考，也希望大家多多支持我們。

本文標(biāo)題: Python ArgumentParse的subparser用法說明

本文地址: http://www.cppcns.com/jiaoben/python/308763.html

總結(jié)

以上是生活随笔為你收集整理的python argvparser_Python ArgumentParse的subparser用法说明的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。