'CON-10+': (2, PandemicSimOpts(use_contact_tracer=True, contact_tracer_history_size=10, spontaneous_testing_rate=0.3)), 'SICK++': (1, PandemicSimOpts(spontaneous_testing_rate=1.)), } param_labels, strategies, sim_opts = zip( *[(k, v[0], v[1]) for k, v in name_to_strategy_sim_opt.items()]) opts = EvaluationOpts(num_seeds=1, strategies=strategies, pandemic_regulations=regulations, sim_opts=sim_opts, max_episode_length=10, enable_warm_up=False) experiment_name = 'contact_tracing' try: evaluate_strategies(experiment_name, opts) except ValueError: # Expect a value error because we are reusing the same directory. pass make_evaluation_plots(exp_name=experiment_name, data_saver_path=opts.data_saver_path, param_labels=param_labels, bar_plot_xlabel='Contact Tracing', show_cumulative_reward=False, annotate_stages=False) plt.show()
experiment_name = name + '_sensitivity' eval_fn(exp_name=experiment_name, eval_opts=opts) except ValueError: # Expect a value error because we are reusing the same directory. pass for name, (eval_fn, param_labels, xlabel) in name_to_eval_fn_labels.items(): experiment_name = name + '_sensitivity' if name == 'social_gatherings': loader = H5DataLoader(experiment_name, path=opts.data_saver_path) data = list(loader.get_data()) data = data[0:1] + data[-2:] param_labels = param_labels[0:1] + param_labels[-2:] make_evaluation_plots_from_data(data=data[:5], exp_name=experiment_name, param_labels=param_labels, bar_plot_xlabel=xlabel, show_summary_plots=False, show_cumulative_reward=False, show_pandemic_duration=False) else: make_evaluation_plots(exp_name=experiment_name, data_saver_path=opts.data_saver_path, param_labels=param_labels, bar_plot_xlabel=xlabel, show_summary_plots=False, show_cumulative_reward=False) plt.show()
if __name__ == '__main__': swedish_strategy = [StageSchedule(stage=0, end_day=3), StageSchedule(stage=1, end_day=None)] italian_strategy = [StageSchedule(stage=0, end_day=3), StageSchedule(stage=1, end_day=8), StageSchedule(stage=2, end_day=13), StageSchedule(stage=3, end_day=25), StageSchedule(stage=4, end_day=59), StageSchedule(stage=3, end_day=79), StageSchedule(stage=2, end_day=None)] opts = EvaluationOpts( num_seeds=30, max_episode_length=180, enable_warm_up=False ) exp_name = 'swedish_italian_strategies' try: eval_government_strategies(exp_name, opts) except ValueError: # Expect a value error because we are reusing the same directory. pass make_evaluation_plots(exp_name=exp_name, data_saver_path=opts.data_saver_path, param_labels=['SWE', 'ITA'], bar_plot_xlabel='Real Government Strategies', annotate_stages=True, show_cumulative_reward=False, show_time_to_peak=False, show_pandemic_duration=True, ) plt.show()
exp_id=1) if __name__ == '__main__': austin_strategy = [StageSchedule(stage=0, end_day=3), StageSchedule(stage=1, end_day=8), StageSchedule(stage=2, end_day=13), StageSchedule(stage=3, end_day=25), StageSchedule(stage=4, end_day=59), StageSchedule(stage=3, end_day=79), StageSchedule(stage=2, end_day=None)] opts = EvaluationOpts( num_seeds=30, max_episode_length=180, enable_warm_up=False ) exp_name = 'bar impact' try: eval_government_strategies(exp_name, opts) except ValueError: pass make_evaluation_plots(exp_name=exp_name, data_saver_path=opts.data_saver_path, param_labels=[' '], bar_plot_xlabel='Bar Impact Strategy', annotate_stages=True, show_cumulative_reward=False, show_time_to_peak=True, show_pandemic_duration=True, ) plt.show()