CS581-Algos-Demo / run_tests.py
Andrei Cozma
Updates
5aaffc4
raw
history blame
453 Bytes
import os
import multiprocessing
vals_eps = [0.5, 0.4, 0.3, 0.2, 0.1, 0.05, 0.01]
vals_gamma = [1.0, 0.99, 0.98, 0.97, 0.95, 0.9, 0.8, 0.7, 0.5]
num_tests = 25
def run_test(args):
os.system(
f"python3 MonteCarloAgent.py --train --gamma {args[0]} --epsilon {args[1]} --wandb_project cs581 --wandb_job_type params"
)
with multiprocessing.Pool(8) as p:
p.map(run_test, [(gamma, eps) for gamma in vals_gamma for eps in vals_eps])