使用argparse运行python程序以在Jupyter Notebook中设置输入

Question

我有一个遵循正常代码结构的.py文件

def  main( args ): 
.......
.......
if __name__ == "__main__":
parser = argparse.ArgumentParser(description = “ forecasting example”)
parser.add_argument("--train-window", default=2160, type=int)
parser.add_argument("--test-window", default=336, type=int)
parser.add_argument("--stride", default=168, type=int)
parser.add_argument("-n", "--num-steps", default=501, type=int)
parser.add_argument("-lr", "--learning-rate", default=0.05, type=float)
parser.add_argument("--dct", action="store_true")
parser.add_argument("--num-samples", default=100, type=int)
parser.add_argument("--log-every", default=50, type=int)
parser.add_argument("--seed", default=1234567890, type=int)
args = parser.parse_args()
main(args)

我试图在Jupyter笔记本上运行这个程序，但它会出现等错误

usage: ipykernel_launcher.py [-h] [--train-window TRAIN_WINDOW]
[--test-window TEST_WINDOW] [--stride STRIDE]
[-n NUM_STEPS] [-lr LEARNING_RATE] [--dct]
[--num-samples NUM_SAMPLES]
[--log-every LOG_EVERY] [--seed SEED]
ipykernel_launcher.py: error: unrecognized arguments: -f C:UsersAppDataRoamingjupyterruntimekernel-4c744f03-3278-4aaf-af5e-50c96e9e41cd.json
An exception has occurred, use %tb to see the full traceback.
SystemExit: 2

我的问题是，如果我想在Jupyter Notebook中运行一个python程序，该程序使用argparse类型的机制设置输入参数，那么正确的方法或修改是什么？

Answer 1

您的代码应该以不同的方式缩进，这样您就可以将其import放入笔记本或另一个Python脚本中。if __name__ == "__main__":块的全部意义在于，当Python解析文件时，它会立即被执行；只有当直接运行文件时，条件才成立，而不是当import时。但是块需要以不同的方式缩进，这样它就不会在任何def、类或其他块结构中。

那么，在笔记本上使用它的方法是使用所需参数调用main(或导入代码中要运行的任何其他函数(。

在这种情况下，main被设计为期望Argparse对象作为其参数，这是非常不幸的。一个更好的设计是简单地在main中进行参数解析，并公开一个不同的函数或一组函数作为库重用。

假设main函数的内部看起来像

def main(args):
print(
real_main(args.train_window, args.test_window,
stride=args.stride, num_steps=args.num_steps,
learning_rate=args.learning_rate,
dct=args.dct, num_samples=args.num_samples,
log_every=args.log_every, seed=args.seed))

假设你想运行

python thatfile.py -n 23 -lr 0.7--dct --num-samples 2300

你笔记本上的等效代码看起来像

from thatfile import real_main as that_main
print(that_main(2160, 336, num_steps=23,
learning_rate=0.7, dct=True,
num_samples=2300))

其中前两个值只是从argparse默认值复制而来，显然我不得不推测哪些参数是必需的，哪些是可选的关键字参数，以及它们的名称是否与argparse字段名称相同。