model deck

import%20marimo%0A%0A__generated_with%20%3D%20%220.23.15%22%0Aapp%20%3D%20marimo.App(width%3D%22full%22)%0A%0A%0A%40app.cell%0Adef%20_()%3A%0A%20%20%20%20import%20marimo%20as%20mo%0A%0A%20%20%20%20return%20(mo%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%20GSF%20Model%20Deck%0A%0A%20%20%20%20The%20standard%20one-look%20summary%20of%20a%20GSF%20model%3A%20flux%20spectrum%2C%20composition%0A%20%20%20%20fractions%2C%20ln%20A%20moments%2C%20relative%20uncertainty%2C%20nucleon%20flux%2C%20and%20the%0A%20%20%20%20all-particle%20flux%20correlation%20across%20energy%20%E2%80%94%20plus%20a%20back-to-back%0A%20%20%20%20comparison%20between%20two%20model%20versions.%20Pick%20the%20versions%20below%3B%20every%0A%20%20%20%20panel%20updates.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Aasync%20def%20_()%3A%0A%20%20%20%20import%20sys%0A%0A%20%20%20%20if%20%22pyodide%22%20in%20sys.modules%3A%0A%20%20%20%20%20%20%20%20%23%20WASM%20(docs%20gallery)%3A%20TEMPORARY%20until%20globalsplinefit%20is%20on%20PyPI%20%E2%80%94%0A%20%20%20%20%20%20%20%20%23%20install%20the%20wheel%20published%20under%20%2Fwheels%2F%20on%20the%20same%20site%20the%0A%20%20%20%20%20%20%20%20%23%20notebook%20is%20served%20from%20(also%20works%20on%20staging%20or%20local%20hosts).%0A%20%20%20%20%20%20%20%20from%20urllib.parse%20import%20urljoin%0A%0A%20%20%20%20%20%20%20%20import%20marimo%20as%20_mo%0A%20%20%20%20%20%20%20%20import%20micropip%0A%0A%20%20%20%20%20%20%20%20%23%20notebook_location()%20resolution%20depth%20differs%20between%20marimo%20export%0A%20%20%20%20%20%20%20%20%23%20layouts%20%E2%80%94%20try%20site-root%20%2Fwheels%2F%20from%20both%20plausible%20depths%0A%20%20%20%20%20%20%20%20_err%20%3D%20None%0A%20%20%20%20%20%20%20%20for%20_up%20in%20(%22..%2F..%2F%22%2C%20%22..%2F..%2F..%2F%22)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20await%20micropip.install(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20urljoin(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20str(_mo.notebook_location())%20%2B%20%22%2F%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20_up%20%2B%20%22wheels%2Fglobalsplinefit-2.0.0a1-py3-none-any.whl%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20_err%20%3D%20None%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20break%0A%20%20%20%20%20%20%20%20%20%20%20%20except%20Exception%20as%20_e%3A%20%20%23%20noqa%3A%20BLE001%20%E2%80%94%20404%20lands%20as%20generic%20error%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20_err%20%3D%20_e%0A%20%20%20%20%20%20%20%20if%20_err%20is%20not%20None%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20raise%20_err%0A%0A%20%20%20%20import%20matplotlib.pyplot%20as%20plt%0A%20%20%20%20import%20numpy%20as%20np%0A%0A%20%20%20%20from%20globalsplinefit%20import%20GSFEnergy%2C%20GSFEnergyPerNucleon%0A%20%20%20%20from%20globalsplinefit.data_management%20import%20MODEL_VERSIONS%0A%0A%20%20%20%20plt.rcParams%5B%22figure.figsize%22%5D%20%3D%20(9%2C%206)%0A%0A%20%20%20%20%23%20TEMPORARY%20pre-publication%20marker%3A%20stamp%20every%20figure%20this%20notebook%0A%20%20%20%20%23%20shows%20with%20a%20diagonal%20PRELIMINARY%20watermark.%20Delete%20at%20the%20GSF%202026%0A%20%20%20%20%23%20release%20(guarded%20so%20a%20cell%20re-run%20does%20not%20stack%20wrappers).%0A%20%20%20%20if%20not%20getattr(plt.show%2C%20%22_gsf_preliminary%22%2C%20False)%3A%0A%20%20%20%20%20%20%20%20_orig_show%20%3D%20plt.show%0A%0A%20%20%20%20%20%20%20%20def%20_show_with_watermark(*args%2C%20**kwargs)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20plt.gcf().text(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%200.5%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%200.5%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%22PRELIMINARY%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20ha%3D%22center%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20va%3D%22center%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20rotation%3D30%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20fontsize%3D42%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20fontweight%3D%22bold%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20color%3D%22gray%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20alpha%3D0.18%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20zorder%3D1000%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20%20%20%20%20return%20_orig_show(*args%2C%20**kwargs)%0A%0A%20%20%20%20%20%20%20%20_show_with_watermark._gsf_preliminary%20%3D%20True%0A%20%20%20%20%20%20%20%20plt.show%20%3D%20_show_with_watermark%0A%20%20%20%20return%20GSFEnergy%2C%20GSFEnergyPerNucleon%2C%20MODEL_VERSIONS%2C%20np%2C%20plt%0A%0A%0A%40app.cell%0Adef%20_(MODEL_VERSIONS%2C%20mo)%3A%0A%20%20%20%20version%20%3D%20mo.ui.dropdown(options%3Dlist(MODEL_VERSIONS)%2C%20value%3D%222026%22%2C%20label%3D%22model%22)%0A%20%20%20%20ref_version%20%3D%20mo.ui.dropdown(%0A%20%20%20%20%20%20%20%20options%3Dlist(MODEL_VERSIONS)%2C%20value%3D%222025%22%2C%20label%3D%22reference%20(comparison)%22%0A%20%20%20%20)%0A%20%20%20%20mo.hstack(%5Bversion%2C%20ref_version%5D%2C%20justify%3D%22start%22)%0A%20%20%20%20return%20ref_version%2C%20version%0A%0A%0A%40app.cell%0Adef%20_(GSFEnergy%2C%20np%2C%20version)%3A%0A%20%20%20%20gsf%20%3D%20GSFEnergy(version%3Dversion.value)%0A%20%20%20%20energy%20%3D%20np.logspace(0%2C%2011%2C%20300)%20%20%23%20GeV%2C%20full%20GSF%20range%0A%20%20%20%20energy_exponent%20%3D%202.6%20%20%23%20E%5E2.6%20flux%20scaling%20for%20display%0A%20%20%20%20scaled%20%3D%20energy**energy_exponent%0A%20%20%20%20groups%20%3D%20%5B%22p%22%2C%20%22He%22%2C%20%22O*%22%2C%20%22Fe*%22%5D%0A%20%20%20%20colors%20%3D%20%7B%0A%20%20%20%20%20%20%20%20%22p%22%3A%20%22%23d62728%22%2C%0A%20%20%20%20%20%20%20%20%22He%22%3A%20%22%23ff7f0e%22%2C%0A%20%20%20%20%20%20%20%20%22O*%22%3A%20%22%232ca02c%22%2C%0A%20%20%20%20%20%20%20%20%22Fe*%22%3A%20%22%231f77b4%22%2C%0A%20%20%20%20%20%20%20%20%22all%22%3A%20%22k%22%2C%0A%20%20%20%20%7D%0A%20%20%20%20return%20colors%2C%20energy%2C%20energy_exponent%2C%20groups%2C%20gsf%2C%20scaled%0A%0A%0A%40app.cell%0Adef%20_(energy%2C%20np)%3A%0A%20%20%20%20def%20flux_err(model%2C%20group)%3A%0A%20%20%20%20%20%20%20%20%22%22%22Flux%20and%201-sigma%20error%20for%20one%20mass%20group%20on%20the%20shared%20grid.%22%22%22%0A%20%20%20%20%20%20%20%20_f%20%3D%20np.asarray(model.flux(energy%2C%20group)%2C%20float)%0A%20%20%20%20%20%20%20%20_e%20%3D%20np.asarray(model.error(energy%2C%20group)%2C%20float)%0A%20%20%20%20%20%20%20%20return%20_f%2C%20_e%0A%0A%20%20%20%20return%20(flux_err%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%20Flux%20spectrum%0A%0A%20%20%20%20All-particle%20flux%20and%20the%20four%20mass%20groups%2C%20scaled%20by%20E%5E2.6%2C%20with%201-sigma%0A%20%20%20%20bands%20propagated%20from%20the%20parameter%20covariance.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(colors%2C%20energy%2C%20flux_err%2C%20groups%2C%20gsf%2C%20np%2C%20plt%2C%20scaled%2C%20version)%3A%0A%20%20%20%20_fig%2C%20_ax%20%3D%20plt.subplots()%0A%20%20%20%20_tf%20%3D%20np.asarray(gsf.total_flux(energy)%2C%20float)%0A%20%20%20%20_te%20%3D%20np.asarray(gsf.total_error(energy)%2C%20float)%0A%20%20%20%20_ax.plot(energy%2C%20_tf%20*%20scaled%2C%20color%3D%22k%22%2C%20lw%3D2.2%2C%20label%3D%22all-particle%22)%0A%20%20%20%20_ax.fill_between(%0A%20%20%20%20%20%20%20%20energy%2C%20(_tf%20-%20_te)%20*%20scaled%2C%20(_tf%20%2B%20_te)%20*%20scaled%2C%20color%3D%22k%22%2C%20alpha%3D0.18%2C%20lw%3D0%0A%20%20%20%20)%0A%20%20%20%20for%20_g%20in%20groups%3A%0A%20%20%20%20%20%20%20%20_f%2C%20_e%20%3D%20flux_err(gsf%2C%20_g)%0A%20%20%20%20%20%20%20%20_ax.plot(energy%2C%20_f%20*%20scaled%2C%20color%3Dcolors%5B_g%5D%2C%20lw%3D1.6%2C%20label%3D_g)%0A%20%20%20%20%20%20%20%20_ax.fill_between(%0A%20%20%20%20%20%20%20%20%20%20%20%20energy%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20(_f%20-%20_e)%20*%20scaled%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20(_f%20%2B%20_e)%20*%20scaled%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20color%3Dcolors%5B_g%5D%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20alpha%3D0.18%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20lw%3D0%2C%0A%20%20%20%20%20%20%20%20)%0A%20%20%20%20_ax.set_xscale(%22log%22)%0A%20%20%20%20_ax.set_yscale(%22log%22)%0A%20%20%20%20_ax.set_ylim(1e2%2C%206e4)%0A%20%20%20%20_ax.set_xlabel(%22E%20%5BGeV%5D%22)%0A%20%20%20%20_ax.set_ylabel(r%22%24E%5E%7B2.6%7D%5C%2C%5CPhi%24%20%5BGeV%24%5E%7B1.6%7D%24m%24%5E%7B-2%7D%24s%24%5E%7B-1%7D%24sr%24%5E%7B-1%7D%24%5D%22)%0A%20%20%20%20_ax.set_title(f%22%7Bversion.value%7D%3A%20flux%20spectrum%20(%C3%97E%24%5E%7B%7B2.6%7D%7D%24%2C%20%C2%B11%CF%83)%22)%0A%20%20%20%20_ax.legend(ncol%3D2%2C%20fontsize%3D9)%0A%20%20%20%20_ax.grid(alpha%3D0.3)%0A%20%20%20%20plt.show()%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%20Composition%20fractions%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(colors%2C%20energy%2C%20groups%2C%20gsf%2C%20plt%2C%20version)%3A%0A%20%20%20%20_fig%2C%20_ax%20%3D%20plt.subplots()%0A%20%20%20%20for%20_g%20in%20groups%3A%0A%20%20%20%20%20%20%20%20_ax.plot(energy%2C%20gsf.fraction(energy%2C%20_g)%2C%20color%3Dcolors%5B_g%5D%2C%20lw%3D1.8%2C%20label%3D_g)%0A%20%20%20%20_ax.set_xscale(%22log%22)%0A%20%20%20%20_ax.set_ylim(0%2C%201)%0A%20%20%20%20_ax.set_xlabel(%22E%20%5BGeV%5D%22)%0A%20%20%20%20_ax.set_ylabel(%22flux%20fraction%22)%0A%20%20%20%20_ax.set_title(f%22%7Bversion.value%7D%3A%20composition%20fractions%22)%0A%20%20%20%20_ax.legend(ncol%3D2%2C%20fontsize%3D9)%0A%20%20%20%20_ax.grid(alpha%3D0.3)%0A%20%20%20%20plt.show()%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%20ln%20A%20moments%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(energy%2C%20gsf%2C%20plt%2C%20version)%3A%0A%20%20%20%20_fig%2C%20_ax%20%3D%20plt.subplots(1%2C%202%2C%20figsize%3D(15%2C%205.5))%0A%20%20%20%20_ax%5B0%5D.plot(energy%2C%20gsf.mean_lnA(energy)%2C%20%22k-%22%2C%20lw%3D1.8)%0A%20%20%20%20_ax%5B0%5D.set_title(f%22%7Bversion.value%7D%3A%20%E2%9F%A8ln%20A%E2%9F%A9%22)%0A%20%20%20%20_ax%5B0%5D.set_ylabel(%22%E2%9F%A8ln%20A%E2%9F%A9%22)%0A%20%20%20%20_ax%5B1%5D.plot(energy%2C%20gsf.var_lnA(energy)%2C%20%22k-%22%2C%20lw%3D1.8)%0A%20%20%20%20_ax%5B1%5D.set_title(f%22%7Bversion.value%7D%3A%20Var(ln%20A)%22)%0A%20%20%20%20_ax%5B1%5D.set_ylabel(%22Var(ln%20A)%22)%0A%20%20%20%20for%20_a%20in%20_ax%3A%0A%20%20%20%20%20%20%20%20_a.set_xscale(%22log%22)%0A%20%20%20%20%20%20%20%20_a.set_xlabel(%22E%20%5BGeV%5D%22)%0A%20%20%20%20%20%20%20%20_a.grid(alpha%3D0.3)%0A%20%20%20%20plt.show()%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%20Relative%20uncertainty%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(colors%2C%20energy%2C%20flux_err%2C%20groups%2C%20gsf%2C%20np%2C%20plt%2C%20version)%3A%0A%20%20%20%20_fig%2C%20_ax%20%3D%20plt.subplots()%0A%20%20%20%20_tf%20%3D%20np.asarray(gsf.total_flux(energy)%2C%20float)%0A%20%20%20%20_te%20%3D%20np.asarray(gsf.total_error(energy)%2C%20float)%0A%20%20%20%20_ax.plot(energy%2C%20_te%20%2F%20_tf%2C%20%22k-%22%2C%20lw%3D2.0%2C%20label%3D%22all-particle%22)%0A%20%20%20%20for%20_g%20in%20groups%3A%0A%20%20%20%20%20%20%20%20_f%2C%20_e%20%3D%20flux_err(gsf%2C%20_g)%0A%20%20%20%20%20%20%20%20_ax.plot(%0A%20%20%20%20%20%20%20%20%20%20%20%20energy%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20_e%20%2F%20np.where(_f%20%3E%200%2C%20_f%2C%20np.nan)%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20color%3Dcolors%5B_g%5D%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20lw%3D1.4%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20label%3D_g%2C%0A%20%20%20%20%20%20%20%20)%0A%20%20%20%20_ax.set_xscale(%22log%22)%0A%20%20%20%20_ax.set_yscale(%22log%22)%0A%20%20%20%20_ax.set_ylim(1e-3%2C%203)%0A%20%20%20%20_ax.set_xlabel(%22E%20%5BGeV%5D%22)%0A%20%20%20%20_ax.set_ylabel(r%22relative%20uncertainty%20%24%5Csigma%2F%5CPhi%24%22)%0A%20%20%20%20_ax.set_title(f%22%7Bversion.value%7D%3A%20relative%20uncertainty%22)%0A%20%20%20%20_ax.legend(ncol%3D2%2C%20fontsize%3D9)%0A%20%20%20%20_ax.grid(alpha%3D0.3)%0A%20%20%20%20plt.show()%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%20Nucleon%20flux%0A%0A%20%20%20%20Proton%20%2B%20neutron%20flux%20per%20energy-per-nucleon%2C%20summed%20over%20all%20nuclei%20%E2%80%94%0A%20%20%20%20the%20quantity%20that%20drives%20atmospheric%20lepton%20production.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(GSFEnergyPerNucleon%2C%20energy_exponent%2C%20np%2C%20plt%2C%20version)%3A%0A%20%20%20%20_nuc%20%3D%20GSFEnergyPerNucleon(version%3Dversion.value)%0A%20%20%20%20_en%20%3D%20np.logspace(0%2C%2011%2C%20300)%0A%20%20%20%20_tot%20%3D%20np.asarray(_nuc.total_flux(_en)%2C%20float)%0A%20%20%20%20_fig%2C%20_ax%20%3D%20plt.subplots()%0A%20%20%20%20_ax.plot(_en%2C%20_tot%20*%20_en**energy_exponent%2C%20%22k-%22%2C%20lw%3D2.0%2C%20label%3D%22nucleon%20(p%2Bn)%22)%0A%20%20%20%20_ax.set_xscale(%22log%22)%0A%20%20%20%20_ax.set_yscale(%22log%22)%0A%20%20%20%20_ax.set_xlabel(%22E%2Fnucleon%20%5BGeV%5D%22)%0A%20%20%20%20_ax.set_ylabel(r%22%24E%5E%7B2.6%7D%5C%2C%5CPhi_%7B%5Crm%20nucleon%7D%24%22)%0A%20%20%20%20_ax.set_title(f%22%7Bversion.value%7D%3A%20nucleon%20flux%22)%0A%20%20%20%20_ax.legend(fontsize%3D9)%0A%20%20%20%20_ax.grid(alpha%3D0.3)%0A%20%20%20%20plt.show()%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%20All-particle%20flux%20correlation%0A%0A%20%20%20%20Correlation%20of%20the%20all-particle%20flux%20between%20energies%2C%20from%20the%20summed%0A%20%20%20%20group-pair%20covariances.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(groups%2C%20gsf%2C%20np%2C%20plt%2C%20version)%3A%0A%20%20%20%20_xc%20%3D%20np.logspace(1%2C%2011%2C%2080)%0A%20%20%20%20_cov%20%3D%20np.zeros((len(_xc)%2C%20len(_xc)))%0A%20%20%20%20for%20_l1%20in%20groups%3A%0A%20%20%20%20%20%20%20%20for%20_l2%20in%20groups%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20_cov%20%2B%3D%20np.asarray(gsf.covariance(_l1%2C%20_l2%2C%20_xc)%2C%20float)%0A%20%20%20%20_d%20%3D%20np.sqrt(np.clip(np.diag(_cov)%2C%201e-300%2C%20None))%0A%20%20%20%20_corr%20%3D%20_cov%20%2F%20np.outer(_d%2C%20_d)%0A%20%20%20%20_fig%2C%20_ax%20%3D%20plt.subplots(figsize%3D(7%2C%206))%0A%20%20%20%20_im%20%3D%20_ax.pcolormesh(%0A%20%20%20%20%20%20%20%20np.log10(_xc)%2C%20np.log10(_xc)%2C%20_corr%2C%20cmap%3D%22RdBu_r%22%2C%20vmin%3D-1%2C%20vmax%3D1%0A%20%20%20%20)%0A%20%20%20%20_fig.colorbar(_im%2C%20ax%3D_ax%2C%20label%3D%22correlation%22)%0A%20%20%20%20_ax.set_xlabel(%22log10%20E%20%5BGeV%5D%22)%0A%20%20%20%20_ax.set_ylabel(%22log10%20E%20%5BGeV%5D%22)%0A%20%20%20%20_ax.set_title(f%22%7Bversion.value%7D%3A%20all-particle%20flux%20correlation%22)%0A%20%20%20%20plt.show()%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo%2C%20ref_version%2C%20version)%3A%0A%20%20%20%20mo.md(rf%22%22%22%0A%20%20%20%20%23%23%20Back-to-back%20comparison%3A%20%7Bversion.value%7D%20vs%20%7Bref_version.value%7D%0A%0A%20%20%20%20Main%20panel%3A%20both%20models%20%C3%97E%5E2.6%20(reference%20solid%2C%20model%20dotted).%0A%20%20%20%20Sub-panels%3A%20per-group%20model%2Freference%20ratio%20(colored%2C%20with%20the%20model's%0A%20%20%20%20error%20band)%20against%20the%20reference's%20own%20relative%20uncertainty%20(grey%20band).%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(%0A%20%20%20%20GSFEnergy%2C%0A%20%20%20%20colors%2C%0A%20%20%20%20energy%2C%0A%20%20%20%20flux_err%2C%0A%20%20%20%20groups%2C%0A%20%20%20%20gsf%2C%0A%20%20%20%20np%2C%0A%20%20%20%20plt%2C%0A%20%20%20%20ref_version%2C%0A%20%20%20%20scaled%2C%0A%20%20%20%20version%2C%0A)%3A%0A%20%20%20%20from%20matplotlib.lines%20import%20Line2D%0A%0A%20%20%20%20_ref%20%3D%20GSFEnergy(version%3Dref_version.value)%0A%20%20%20%20_items%20%3D%20groups%20%2B%20%5B%22all%22%5D%0A%0A%20%20%20%20def%20_fe(model%2C%20group)%3A%0A%20%20%20%20%20%20%20%20if%20group%20%3D%3D%20%22all%22%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20return%20(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20np.asarray(model.total_flux(energy)%2C%20float)%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20np.asarray(model.total_error(energy)%2C%20float)%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20return%20flux_err(model%2C%20group)%0A%0A%20%20%20%20_fn%20%3D%20%7B_g%3A%20_fe(gsf%2C%20_g)%20for%20_g%20in%20_items%7D%0A%20%20%20%20_fr%20%3D%20%7B_g%3A%20_fe(_ref%2C%20_g)%20for%20_g%20in%20_items%7D%0A%0A%20%20%20%20_fig%20%3D%20plt.figure(figsize%3D(8.5%2C%2011))%0A%20%20%20%20_gs%20%3D%20_fig.add_gridspec(6%2C%201%2C%20height_ratios%3D%5B3.2%2C%201%2C%201%2C%201%2C%201%2C%201%5D%2C%20hspace%3D0.12)%0A%20%20%20%20_ax0%20%3D%20_fig.add_subplot(_gs%5B0%5D)%0A%20%20%20%20_rax%20%3D%20%5B_fig.add_subplot(_gs%5B_i%20%2B%201%5D%2C%20sharex%3D_ax0)%20for%20_i%20in%20range(5)%5D%0A%20%20%20%20for%20_g%20in%20_items%3A%0A%20%20%20%20%20%20%20%20_c%20%3D%20colors%5B_g%5D%0A%20%20%20%20%20%20%20%20_ax0.plot(energy%2C%20_fr%5B_g%5D%5B0%5D%20*%20scaled%2C%20_c%2C%20ls%3D%22-%22%2C%20lw%3D1.8)%0A%20%20%20%20%20%20%20%20_ax0.fill_between(%0A%20%20%20%20%20%20%20%20%20%20%20%20energy%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20(_fr%5B_g%5D%5B0%5D%20-%20_fr%5B_g%5D%5B1%5D)%20*%20scaled%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20(_fr%5B_g%5D%5B0%5D%20%2B%20_fr%5B_g%5D%5B1%5D)%20*%20scaled%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20color%3D_c%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20alpha%3D0.18%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20lw%3D0%2C%0A%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20_ax0.plot(energy%2C%20_fn%5B_g%5D%5B0%5D%20*%20scaled%2C%20_c%2C%20ls%3D%22%3A%22%2C%20lw%3D1.8)%0A%20%20%20%20_ax0.set_xscale(%22log%22)%0A%20%20%20%20_ax0.set_yscale(%22log%22)%0A%20%20%20%20_ax0.set_ylim(1e2%2C%206e4)%0A%20%20%20%20_ax0.set_ylabel(r%22%24E%5E%7B2.6%7D%5C%2C%5CPhi%24%22)%0A%20%20%20%20_ax0.set_title(%0A%20%20%20%20%20%20%20%20f%22Back-to-back%3A%20%7Bref_version.value%7D%20(solid)%20vs%20%7Bversion.value%7D%20(dotted)%22%0A%20%20%20%20)%0A%20%20%20%20_leg%20%3D%20%5BLine2D(%5B%5D%2C%20%5B%5D%2C%20color%3Dcolors%5B_g%5D%2C%20label%3D_g)%20for%20_g%20in%20_items%5D%0A%20%20%20%20_leg%20%2B%3D%20%5B%0A%20%20%20%20%20%20%20%20Line2D(%5B%5D%2C%20%5B%5D%2C%20color%3D%22gray%22%2C%20ls%3D%22-%22%2C%20label%3Dref_version.value)%2C%0A%20%20%20%20%20%20%20%20Line2D(%5B%5D%2C%20%5B%5D%2C%20color%3D%22gray%22%2C%20ls%3D%22%3A%22%2C%20label%3Dversion.value)%2C%0A%20%20%20%20%5D%0A%20%20%20%20_ax0.legend(handles%3D_leg%2C%20ncol%3D2%2C%20fontsize%3D8)%0A%20%20%20%20for%20_ax%2C%20_g%20in%20zip(_rax%2C%20_items%2C%20strict%3DTrue)%3A%0A%20%20%20%20%20%20%20%20_c%20%3D%20colors%5B_g%5D%0A%20%20%20%20%20%20%20%20(_fnf%2C%20_fne)%2C%20(_frf%2C%20_fre)%20%3D%20_fn%5B_g%5D%2C%20_fr%5B_g%5D%0A%20%20%20%20%20%20%20%20with%20np.errstate(divide%3D%22ignore%22%2C%20invalid%3D%22ignore%22)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20_ratio%20%3D%20_fnf%20%2F%20np.where(_frf%20%3E%200%2C%20_frf%2C%20np.nan)%0A%20%20%20%20%20%20%20%20%20%20%20%20_rel%20%3D%20_fre%20%2F%20np.where(_frf%20%3E%200%2C%20_frf%2C%20np.nan)%0A%20%20%20%20%20%20%20%20%20%20%20%20_band%20%3D%20_fne%20%2F%20np.where(_frf%20%3E%200%2C%20_frf%2C%20np.nan)%0A%20%20%20%20%20%20%20%20_ax.fill_between(energy%2C%201%20-%20_rel%2C%201%20%2B%20_rel%2C%20color%3D%220.8%22%2C%20lw%3D0)%0A%20%20%20%20%20%20%20%20_ax.plot(energy%2C%20_ratio%2C%20_c%2C%20lw%3D1.6)%0A%20%20%20%20%20%20%20%20_ax.fill_between(%0A%20%20%20%20%20%20%20%20%20%20%20%20energy%2C%20_ratio%20-%20_band%2C%20_ratio%20%2B%20_band%2C%20color%3D_c%2C%20alpha%3D0.25%2C%20lw%3D0%0A%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20_ax.axhline(1.0%2C%20color%3D%22k%22%2C%20lw%3D0.6)%0A%20%20%20%20%20%20%20%20_ax.set_ylabel(f%22%7B_g%7D%5Cnnew%2Fref%22%2C%20fontsize%3D8)%0A%20%20%20%20%20%20%20%20_ax.set_ylim(0.6%2C%201.4)%0A%20%20%20%20%20%20%20%20_ax.set_xscale(%22log%22)%0A%20%20%20%20%20%20%20%20_ax.grid(alpha%3D0.25)%0A%20%20%20%20%20%20%20%20for%20_t%20in%20_ax.get_xticklabels()%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20_t.set_visible(False)%0A%20%20%20%20for%20_t%20in%20_rax%5B-1%5D.get_xticklabels()%3A%0A%20%20%20%20%20%20%20%20_t.set_visible(True)%0A%20%20%20%20_rax%5B-1%5D.set_xlabel(%22E%20%5BGeV%5D%22)%0A%20%20%20%20plt.show()%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%20Custom%20fits%0A%0A%20%20%20%20Every%20panel%20works%20the%20same%20for%20a%20parameter%20directory%20that%20is%20not%20in%20the%0A%20%20%20%20registry%20%E2%80%94%20construct%20the%20model%20with%20%60GSFEnergy(data_path%3D%22my_fit%2F%22)%60%0A%20%20%20%20instead%20of%20%60version%3D...%60%20(run%20the%20notebook%20locally%20for%20that%3B%20the%20browser%0A%20%20%20%20version%20has%20no%20filesystem%20access).%20To%20save%20any%20figure%2C%20use%0A%20%20%20%20%60fig.savefig(%22name.png%22%2C%20dpi%3D120%2C%20bbox_inches%3D%22tight%22)%60%20in%20a%20cell.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0Aif%20__name__%20%3D%3D%20%22__main__%22%3A%0A%20%20%20%20app.run()%0A