Jupyter Notebook運(yùn)行Python代碼實(shí)現(xiàn)傳參方式
Jupyter Notebook運(yùn)行Python代碼實(shí)現(xiàn)傳參
在Jupyter Notebook中,運(yùn)行Python源代碼非常方便,但是如何模擬命令行方式運(yùn)行時(shí)的輸入?yún)?shù)呢?
如果直接使用sys.argv會(huì)出現(xiàn)錯(cuò)誤。
例如,使用argv[1]時(shí)出現(xiàn)

導(dǎo)致上述錯(cuò)誤的原因?yàn)椋涸贘upyter Notebook中運(yùn)行Python代碼時(shí),argv中有三個(gè)默認(rèn)參數(shù),分別是:
argv[0]: ‘路徑\test.py' (源代碼文件名) argv[1]: ‘-f' argv[2]: '路徑\kernel-…json'
而在命令行方式下運(yùn)行Python代碼時(shí),argv[0]與上述argv[0]相同,但argv[1]為命令行中緊隨在test.py之后的字符串
例如,在如下命令行中
python test.py test.txt argv[1]為 test.txt
因此,解決方案就是對(duì) list argv 進(jìn)行修改,如下面的代碼所示:
#需要sys模塊來(lái)使用argv
from sys import argv
#查看當(dāng)前的argv列表
print(len(argv))
print(argv)
#直接修改argv
argv[1] = 'test.txt'
#也可以添加新的參數(shù)
argv.append('test.txt')這樣,在后續(xù)代碼中, .py 和 .ipynb源代碼一致,無(wú)需為在Jupyter Notebook中運(yùn)行而進(jìn)行修改。
jupyter notebook參數(shù)化運(yùn)行python
Updates
(2019.8.14 19:53)吃飯前用這個(gè)方法實(shí)戰(zhàn)了一下,吃完回來(lái)一看好像不太行:跑完一組參數(shù)之后,到跑下一組參數(shù)時(shí)好像沒(méi)有釋放之占用的 GPU,于是 notebook 上的結(jié)果,后面好幾條都報(bào)錯(cuò)說(shuō) cuda out of memory。
現(xiàn)在改成:將 notebook 中的代碼寫(xiě)在一個(gè) python 文件中,然后用命令行運(yùn)行這個(gè)文件,比如:
# autorun.py
import os
# print(os.getcwd())
over = [ ?# 之前手工改參數(shù)跑完的參數(shù)組合
? ? [0, 1, 1], [0, 1, 2], [0, 1, 3],
? ? [0, 2, 1],
? ? [1, 0, 1],
? ? [1, 2, 1]
]
for alpha in range(1, 4, 1):
? ? for beta in range(3):
? ? ? ? for gamma in range(3):
? ? ? ? ? ? if [alpha, beta, gamma] in over:
? ? ? ? ? ? ? ? continue
? ? ? ? ? ? os.system(f'python main.py --alpha {alpha} --beta {beta} --gamma {gamma}')這里的 main.py 是訓(xùn)練用的主文件。改在 py 里用 os.system 跑,希望跑一組參數(shù)之后完會(huì)自動(dòng)釋放資源再跑下一組(?)
Notes
有多組參數(shù)組合需要嘗試,不想每組參數(shù)都人工修改 python 代碼,再在 notebook 中 %run 它。
python 參數(shù)通過(guò)的 argparse 接收,在 notebook 中寫(xiě)個(gè)多重循環(huán)遍歷參數(shù)組合傳給 python 程序自動(dòng)運(yùn)行。
記錄一個(gè)簡(jiǎn)例。
Codes
test_dir
|- test.py
|- test.ipynb
in py file
# test.py
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--number', type=int, default=0, help='number')
parser.add_argument('--string', type=str, default='abc', help='string')
args = parser.parse_args()
print('number:', args.number, type(args.number))
print('string:', args.string, type(args.string))in notebook
注意傳參數(shù)時(shí) $ 的使用
# test.ipynb
for i in range(3):
? ? for s in ('a', 'b', 'c'):
? ? ? ? %run test.py --number $i --string $s總結(jié)
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
Python?logging日志模塊的概念與實(shí)踐講解
本文通過(guò)具體的代碼示例為大家解釋了如何高效地使用logging模塊進(jìn)行日志記錄,以及如何避免常見(jiàn)的陷阱,希望可以幫助大家更好地掌握這個(gè)強(qiáng)大的工具2023-07-07
使用pyqt5搭建yolo3目標(biāo)識(shí)別界面的方法
這篇文章主要介紹了使用pyqt5搭建yolo3目標(biāo)識(shí)別界面的方法,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-03-03
Tensorflow使用tfrecord輸入數(shù)據(jù)格式
這篇文章主要介紹了Tensorflow使用tfrecord輸入數(shù)據(jù)格式,小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2018-06-06
Python實(shí)現(xiàn)截取PDF文件中的幾頁(yè)代碼實(shí)例
今天小編就為大家分享一篇關(guān)于Python實(shí)現(xiàn)截取PDF文件中的幾頁(yè)代碼實(shí)例,小編覺(jué)得內(nèi)容挺不錯(cuò)的,現(xiàn)在分享給大家,具有很好的參考價(jià)值,需要的朋友一起跟隨小編來(lái)看看吧2019-03-03
Python操作系統(tǒng)的6個(gè)自動(dòng)化腳本小結(jié)
在Python中,實(shí)現(xiàn)操作系統(tǒng)自動(dòng)化的腳本可以涵蓋從文件操作、系統(tǒng)監(jiān)控到網(wǎng)絡(luò)任務(wù)等多種功能,下面我將詳細(xì)介紹六個(gè)不同類(lèi)別的Python自動(dòng)化腳本示例,這些示例將幫助你理解如何用Python來(lái)自動(dòng)化日常操作系統(tǒng)任務(wù),需要的朋友可以參考下2024-10-10
Python中__new__與__init__方法的區(qū)別詳解
這篇文章主要介紹了Python中__new__與__init__方法的區(qū)別,是Python學(xué)習(xí)中的基礎(chǔ)知識(shí),需要的朋友可以參考下2015-05-05

