测试不同参数时,同一个模型需要运行多次,此时需要输入不同的参数观察结果。编写一个控制程序,自动输入不同参数执行python文件,无需手动执行每一个。
add.py:需要重复执行的程序
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--number1', type=int, default=0, help='number1')
parser.add_argument('--number2', type=int, default=0, help='number2')
args = parser.parse_args()
print('{} + {} = {}'.format(args.number1, args.number2, args.number1 + args.number2))
main.py:用于控制add.py的程序,可以指定参数,与add.py放在相同路径上
import os
for a in [1, 2, 3, 4, 5]:
for b in [10, 20, 30]:
os.system("python add.py --number1 {} --number2 {}".format(a, b))
结果
'''
1 + 10 = 11
1 + 20 = 21
1 + 30 = 31
2 + 10 = 12
2 + 20 = 22
2 + 30 = 32
3 + 10 = 13
3 + 20 = 23
3 + 30 = 33
4 + 10 = 14
4 + 20 = 24
4 + 30 = 34
5 + 10 = 15
5 + 20 = 25
5 + 30 = 35
'''
网友评论