def defend_the_center(name="center_def"): defaults, net_defaults = _default_engine_args() custom_args = { "name": name, "config_file": "config/defend_the_center.cfg" } network_args = net_defaults engine_args = _merge_dicts(defaults, custom_args) return QEngine(network_args=network_args, **engine_args)
def predict_supreme(name="predict-s_def"): defaults, net_defaults = _default_engine_args() custom_args = { "name": name, "config_file": "config/predict_position_supreme.cfg" } network_args = net_defaults engine_args = _merge_dicts(defaults, custom_args) return QEngine(network_args=network_args, **engine_args)
def pacman(name="pacman_def"): defaults, net_defaults = _default_engine_args() custom_args = { "name": name, "config_file": "config/pacman.cfg" } network_args = net_defaults engine_args = _merge_dicts(defaults, custom_args) return QEngine(network_args=network_args, **engine_args)
def take_cover_simple(name="cover_simple_def"): defaults, net_defaults = _default_engine_args() custom_args = { "name": name, "config_file": "config/take_cover_simple.cfg" } network_args = net_defaults engine_args = _merge_dicts(defaults, custom_args) return QEngine(network_args=network_args, **engine_args)
def health(name="health_def_skip7"): defaults, net_defaults = _default_engine_args() custom_args = { "name": name, "config_file": "config/health_gathering.cfg", "skiprate": 7, } network_args = net_defaults engine_args = _merge_dicts(defaults, custom_args) return QEngine(network_args=network_args, **engine_args)
def health_baseline(name="health_baseline4"): defaults, net_defaults = _default_engine_args() custom_args = { "name": name, "config_file": "config/health_gathering.cfg", "skiprate": 9, "count_time": False, "reshaped_x": 84, "reshaped_y": 84, "net_type": "dqn", } network_args = net_defaults net_defaults["gamma"] = 0.99 engine_args = _merge_dicts(defaults, custom_args) return QEngine(network_args=network_args, **engine_args)
from tqdm import trange my_range = trange if args.list_agents: print "Available agents in agents.py:" for member in getmembers(agents): if isfunction(member[1]): if member[1].__name__[0] != "_": print " ", member[1].__name__ exit(0) results = None if args.agent_file[0]: engine = QEngine.load(args.agent_file[0], config_file=args.config_fil1[0]) results = pickle.load(open(engine.results_file, "r")) else: if args.agent is not None: if args.name[0] is not None: engine = getattr(agents, args.agent)(args.name[0]) else: engine = getattr(agents, args.agent)() elif args.json_file[0] is not None: engine_args = yaml.load(open(args.json_file[0], "r")) if args.name[0] is not None: engine_args["name"] = args.name[0] if args.config_file[0] is not None: engine_args["config_file"] = args.config_file[0] engine = QEngine(**engine_args)
else: from tqdm import trange my_range = trange if args.list_agents: print "Available agents in agents.py:" for member in getmembers(agents): if isfunction(member[1]): if member[1].__name__[0] != "_": print " ", member[1].__name__ exit(0) results = None if args.agent_file[0]: engine = QEngine.load(args.agent_file[0], config_file=args.config_fil1[0]) results = pickle.load(open(engine.results_file, "r")) else: if args.agent is not None: if args.name[0] is not None: engine = getattr(agents, args.agent)(args.name[0]) else: engine = getattr(agents, args.agent)() elif args.json_file[0] is not None: engine_args = yaml.load(open(args.json_file[0], "r")) if args.name[0] is not None: engine_args["name"] = args.name[0] if args.config_file[0] is not None: engine_args["config_file"] = args.config_file[0] engine = QEngine(**engine_args)