generate_plots.py

#!/usr/bin/env python3

import plotting
import numpy as np

# --- PLOTTING DATA: DEATH MODEL --- #

data = np.load('./deathmodel_dim1.npz')
filename = 'deathmodel_utility'

# UTILITY
if type(data['bo_obj']) == np.ndarray:
    obj = data['bo_obj'].tolist()
else:
    obj = data['bo_obj']
plotting.plot_acquisition(obj, data['d_opt'], filename=filename)

# POSTERIOR
pp = data['prior_samples']
rr = data['r_obs']
plotting.plot_posterior(pp, rr, model='death', truth=1.5)

# --- PLOTTING DATA: SIR MODEL --- #

data = np.load('./sirmodel_dim1.npz')
filename = 'sirmodel_utility'

# UTILITY
if type(data['bo_obj']) == np.ndarray:
    obj = data['bo_obj'].tolist()
else:
    obj = data['bo_obj']
plotting.plot_acquisition(obj, data['d_opt'], filename=filename)

# POSTERIOR
pp = data['prior_samples']
rr = data['r_obs']
plotting.plot_posterior(pp, rr, model='sir', truth=np.array([0.15, 0.05]))