diff --git a/fitting/ModelFit.py b/fitting/ModelFit.py index 64d58eb..e093d99 100644 --- a/fitting/ModelFit.py +++ b/fitting/ModelFit.py @@ -159,11 +159,17 @@ class ModelFit: step = (max_x - min_x) / 5000 x_values = np.arange(min_x, max_x + step, step) + + # plot baseline f_base_color = ("blue", "deepskyblue") f_inf_color = ("green", "limegreen") f_zero_color = ("red", "orange") + # plot EOD frequency in fi-plot + eod_freq = cell.get_eod_frequency() + axes[1].plot((min_x, max_x), (eod_freq, eod_freq), color="black", label="eod freq") + median_baseline = np.median(fi_curve_cell.get_f_baseline_frequencies()) axes[1].plot((min_x, max_x), (median_baseline, median_baseline), color=f_base_color[0], label="cell med base") axes[1].plot(fi_curve_model.stimulus_values, fi_curve_model.get_f_baseline_frequencies(),