Andrei Cozma commited on
Commit
401880e
·
1 Parent(s): ec64bb9
Files changed (1) hide show
  1. run_tests.py +7 -2
run_tests.py CHANGED
@@ -9,9 +9,14 @@ num_tests = 25
9
 
10
  def run_test(args):
11
  os.system(
12
- f"python3 MonteCarloAgent.py --train --gamma {args[0]} --epsilon {args[1]} --wandb_project cs581 --wandb_job_type params"
13
  )
14
 
15
 
16
  with multiprocessing.Pool(8) as p:
17
- p.map(run_test, [(gamma, eps) for gamma in vals_gamma for eps in vals_eps])
 
 
 
 
 
 
9
 
10
  def run_test(args):
11
  os.system(
12
+ f"python3 MonteCarloAgent.py --train --gamma {args[0]} --epsilon {args[1]} --wandb_project cs581 --wandb_job_type params --wandb_run_name_suffix {args[2]}"
13
  )
14
 
15
 
16
  with multiprocessing.Pool(8) as p:
17
+ # make all the tests
18
+ tests = []
19
+ for gamma in vals_gamma:
20
+ for eps in vals_eps:
21
+ tests.extend((gamma, eps, i) for i in range(num_tests))
22
+ p.map(run_test, tests)