Skip to content
Snippets Groups Projects
Commit 4ea23699 authored by Antonello, Dr. Massimiliano's avatar Antonello, Dr. Massimiliano
Browse files

Added the prefit_only oiption in sumlab_auto

parent 068539d7
No related branches found
No related tags found
No related merge requests found
...@@ -3,3 +3,5 @@ data ...@@ -3,3 +3,5 @@ data
.DS_Store .DS_Store
__pycache__ __pycache__
user user
Icon*
peakotron_logo.icns
\ No newline at end of file
...@@ -85,12 +85,10 @@ for i, (file, path) in enumerate(files_to_fit): ...@@ -85,12 +85,10 @@ for i, (file, path) in enumerate(files_to_fit):
tau_R=f_tau_hmt*tau, tau_R=f_tau_hmt*tau,
bin_0=bin_0, bin_0=bin_0,
truncate_nsigma0_up=truncate_nsigma0_up, truncate_nsigma0_up=truncate_nsigma0_up,
truncate_nsigma0_do=truncate_nsigma0_do truncate_nsigma0_do=truncate_nsigma0_do,
prefit_only=prefit_only
) )
f_data.PlotFit(plot_in_bins=True, display=False,
save_directory=f"{folder}/{file[:-4]}_fit.png")
fit_out = {} fit_out = {}
prefit_val, prefit_err = f_data.GetPrefitResults(bin_units=False) prefit_val, prefit_err = f_data.GetPrefitResults(bin_units=False)
print("\033[95m"+rf"Prefit: G = {prefit_val.get('G')} d_G = {prefit_err.get('G')}"+"\033[0m") print("\033[95m"+rf"Prefit: G = {prefit_val.get('G')} d_G = {prefit_err.get('G')}"+"\033[0m")
...@@ -99,13 +97,15 @@ for i, (file, path) in enumerate(files_to_fit): ...@@ -99,13 +97,15 @@ for i, (file, path) in enumerate(files_to_fit):
for key, value in prefit_err.items(): for key, value in prefit_err.items():
fit_out["prefit_d_{:s}".format(key)] = value fit_out["prefit_d_{:s}".format(key)] = value
#if not prefit_only: if not prefit_only:
fit_val, fit_err = f_data.GetFitResults(bin_units=False) fit_val, fit_err = f_data.GetFitResults(bin_units=False)
print("\033[95m"+rf"Fit: G = {fit_val.get('G')} d_G = {fit_err.get('G')}"+"\033[0m") print("\033[95m"+rf"Fit: G = {fit_val.get('G')} d_G = {fit_err.get('G')}"+"\033[0m")
for key, value in fit_val.items(): for key, value in fit_val.items():
fit_out["{:s}".format(key)] = value fit_out["{:s}".format(key)] = value
for key, value in fit_err.items(): for key, value in fit_err.items():
fit_out["d_{:s}".format(key)] = value fit_out["d_{:s}".format(key)] = value
f_data.PlotFit(plot_in_bins=True, display=False,
save_directory=f"{folder}/{file[:-4]}_fit.png")
df = pd.DataFrame.from_dict([fit_out]) df = pd.DataFrame.from_dict([fit_out])
df.to_csv("{}/fit_results_{:s}.csv".format(folder, file[:-4])) df.to_csv("{}/fit_results_{:s}.csv".format(folder, file[:-4]))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment