increase f_zero points weight

This commit is contained in:
alexanderott 2021-01-10 00:08:40 +01:00
parent ea580863b1
commit ad7a8bcfd3
2 changed files with 6 additions and 4 deletions

View File

@ -235,6 +235,8 @@ class ModelFit:
if save_path is None: if save_path is None:
plt.show() plt.show()
else: else:
if not os.path.exists(save_path):
os.makedirs(save_path)
plt.savefig(save_path + cell.get_cell_name() + "_master_plot.pdf") plt.savefig(save_path + cell.get_cell_name() + "_master_plot.pdf")
plt.close() plt.close()

View File

@ -14,13 +14,13 @@ from my_util.helperFunctions import plot_errors
import multiprocessing as mp import multiprocessing as mp
SAVE_DIRECTORY = "./results/final_sam/" SAVE_DIRECTORY = "./results/final_sam2/"
SAVE_DIRECTORY_BEST = "./results/final_sam_best/" # SAVE_DIRECTORY_BEST = "./results/final_sam2_best/"
# SAVE_DIRECTORY = "./results/ref_and_tau/no_dend_tau/" # SAVE_DIRECTORY = "./results/ref_and_tau/no_dend_tau/"
# SAVE_DIRECTORY_BEST = "./results/ref_and_tau/ndt_best/" # SAVE_DIRECTORY_BEST = "./results/ref_and_tau/ndt_best/"
# [vs, sc, cv, isi_hist, bursty, f_inf, f_inf_slope, f_zero, f_zero_slope, f0_curve] # [vs, sc, cv, isi_hist, bursty, f_inf, f_inf_slope, f_zero, f_zero_slope, f0_curve]
ERROR_WEIGHTS = (1, 1, 1, 1, 1, 1, 1, 1, 0, 1) ERROR_WEIGHTS = (1, 1, 1, 1, 1, 1, 1, 10, 0, 1)
def main(): def main():
@ -108,7 +108,7 @@ def fit_cell_parallel(cell_data, start_parameters):
del cell_data del cell_data
best_fit = get_best_fit(save_path_cell) best_fit = get_best_fit(save_path_cell)
best_fit.generate_master_plot(SAVE_DIRECTORY_BEST) # best_fit.generate_master_plot(SAVE_DIRECTORY_BEST)
best_fit.generate_master_plot(SAVE_DIRECTORY) best_fit.generate_master_plot(SAVE_DIRECTORY)