首先需要import sys,sys是python3的一个标准库 / 模块;
「argv」是「argument variable」参数变量的简写形式;是一个List列表。
argv[0] 一般是被调用的脚本文件名或全路径,和操作系统有关。 sys.argv[0]表示代码本身文件路径,
所以参数从1开始:argv[1]及之后是传入的数据。
下面这个例子中,我们先编写一个run_cnn.py文件,并定义好train 、test函数,然后再编写main函数,这样的话,我们就可以在terminal中分别输入 python run_cnn.py [train / test]来选择是训练还是测试模型
if __name__ == '__main__':
if len(sys.argv) != 2 or sys.argv[1] not in ['train', 'test']:
raise ValueError("""usage: python run_cnn.py [train / test]""")
print('Configuring CNN model...')
''' 模型配置'''
if sys.argv[1] == 'train':
train()
else:
test()