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)