def experiment(work_dir, problem_cls=None, solver_cls=None, num_rounds=None, batch_size=None, seed=None): """Run experiment.""" # Must be specified via Gin. assert problem_cls assert solver_cls assert num_rounds assert batch_size tf.gfile.MakeDirs(work_dir) if seed is not None: utils.set_seed(seed) print('Running experiment with %s on %s' % (problem_cls, solver_cls)) problem = problem_cls() solver = solver_cls(problem.domain) population = controller.run(problem, solver, num_rounds=num_rounds, batch_size=batch_size) print('Writing output to %s/population.csv' % work_dir) population.to_csv(os.path.join(work_dir, 'population.csv')) return population
def test_solve(self): problem = simple_ising_model.AlternatingChainIsingModel(length=8, vocab_size=4) solver = deep_evolution_solver.MutationPredictorSolver( domain=problem.domain) controller.run(problem, solver, num_rounds=20, batch_size=10)