def _setup_experiment(): return attention_allocation_experiment.Experiment( num_runs=1, num_steps=10, seed=0, env_class=attention_allocation.LocationAllocationEnv, env_params=_get_base_env_params(), agent_class=random_agents.RandomAgent)
def _setup_experiment(): return attention_allocation_experiment.Experiment( num_runs=50, num_steps=1000, num_workers=25, seed=0, env_class=attention_allocation.LocationAllocationEnv, env_params=_get_base_env_params())