DooABLe / assets /figure1_source /make_diagnostics.py
pranamanam's picture
Upload 309 files
81ae663 verified
Raw
History Blame Contribute Delete
1.48 kB
"""Empty experimental axes. No simulated curves or invented measurements."""
from pathlib import Path
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
from matplotlib import font_manager
root=Path(__file__).resolve().parent
font_manager.fontManager.addfont(str(root/'fonts/Ubuntu-Regular.ttf'))
plt.rcParams.update({'font.family':'Ubuntu','font.size':9,'mathtext.fontset':'cm','axes.spines.top':False,'axes.spines.right':False,'axes.labelcolor':'#243644','axes.edgecolor':'#91A3AF','xtick.color':'#526C7D','ytick.color':'#526C7D','pdf.fonttype':42})
fig,axs=plt.subplots(3,1,figsize=(3.25,3.9))
spec=[('Route multiplicity',r'$P(Y=y_1)$',[1,4,8,16],(0,1)),('Route temperature',r'$\mathbb{E}[C\mid Y]$',[.1,1,2],None),('Training steps','Endpoint TV',[0,1000,2000],(0,1))]
for ax,(xl,yl,ticks,ylim) in zip(axs,spec):
ax.set_xlabel(xl);ax.set_ylabel(yl);ax.set_xticks(ticks)
if ylim:ax.set_ylim(*ylim)
else:ax.set_yticks([])
fig.tight_layout(pad=.5,h_pad=1.0);fig.savefig(root/'diagnostics.pdf');fig.savefig(root/'diagnostics.png',dpi=200)
fig,axs=plt.subplots(2,1,figsize=(3.25,2.8))
axs[0].set(xlabel='Affinity utility',ylabel='Permeability utility',xlim=(0,1),ylim=(0,1))
axs[1].set(xlabel='Parent compounds',ylabel='Time (s)',xscale='log',xlim=(100,10000))
axs[1].set_xticks([100,1000,10000],['100','1,000','10,000'])
axs[1].set_yticks([])
fig.tight_layout(pad=.6,h_pad=1.0)
fig.savefig(root/'preferences.pdf')
fig.savefig(root/'preferences.png',dpi=200)