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)