diff --git a/pes_to_spec/test/offline_analysis.py b/pes_to_spec/test/offline_analysis.py index 5944147c602de651535eda6bbbede97e54dfd9c8..d1191bfae3643e3af4ca8cf0821f0a8d4c95c380 100755 --- a/pes_to_spec/test/offline_analysis.py +++ b/pes_to_spec/test/offline_analysis.py @@ -70,18 +70,20 @@ def plot_result(filename: str, spec_pred: Dict[str, np.ndarray], spec_smooth: np ax = fig.add_subplot(gs[0, 0]) unc_stat = np.mean(spec_pred["unc"]) unc_pca = np.mean(spec_pred["pca"]) - #ax.plot(spec_raw_pe, spec_smooth, c='b', lw=3, label="High-resolution measurement (smoothened)") + ax.plot(spec_raw_pe, spec_smooth, c='b', lw=3, label="High-resolution measurement (smoothened)") ax.plot(spec_raw_pe, spec_pred["expected"], c='r', lw=3, label="High-resolution prediction") unc = np.sqrt(spec_pred["unc"]**2 + spec_pred["pca"]**2) ax.fill_between(spec_raw_pe, spec_pred["expected"] - unc, spec_pred["expected"] + unc, facecolor='red', alpha=0.6, label="68% unc.") #ax.fill_between(spec_raw_pe, spec_pred["expected"] - spec_pred["unc"], spec_pred["expected"] + spec_pred["unc"], facecolor='red', alpha=0.6, label="68% unc. (stat.)") #ax.fill_between(spec_raw_pe, spec_pred["expected"] - spec_pred["pca"], spec_pred["expected"] + spec_pred["pca"], facecolor='magenta', alpha=0.6, label="68% unc. (syst., PCA)") - if spec_raw_int is not None: - ax.plot(spec_raw_pe, spec_raw_int, c='b', lw=1, ls='--', label="High-resolution measurement") + #if spec_raw_int is not None: + # ax.plot(spec_raw_pe, spec_raw_int, c='b', lw=1, ls='--', label="High-resolution measurement") ax.legend(frameon=False, borderaxespad=0) + Y = np.amax(spec_smooth) ax.set(title=f"", #avg(stat unc) = {unc_stat}, avg(pca unc) = {unc_pca}", xlabel="Photon energy [eV]", - ylabel="Intensity") + ylabel="Intensity", + ylim=(0, 1.3*Y)) fig.savefig(filename) plt.close(fig)