Alexander/P-unit_model
2
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add plotting and saving of f point detection

Browse Source
This commit is contained in:
a.ott 2020-05-18 12:19:11 +02:00
parent a41de94961
commit d086f4b3a1
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
variableEffect.py
View File

@ -5,6 +5,7 @@ from FiCurve import FICurveModel
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import copy import copy
import os
SEARCH_WIDTH = 3 SEARCH_WIDTH = 3
@ -66,6 +67,16 @@ def test_parameter_effect(model_parameters, test_parameter):
f_zero_s.append(fi_curve.get_f_zero_fit_slope_at_stimulus_value(0.1)) f_zero_s.append(fi_curve.get_f_zero_fit_slope_at_stimulus_value(0.1))
f_zero_v.append(fi_curve.get_f_zero_frequencies()) f_zero_v.append(fi_curve.get_f_zero_frequencies())
if not os.path.exists("./figures/f_point_detection/"):
os.makedirs("./figures/f_point_detection/")
detection_save_path = "./figures/f_point_detection/{}_{:.4f}/".format(test_parameter, values[i])
if not os.path.exists(detection_save_path):
os.makedirs(detection_save_path)
fi_curve.plot_f_point_detections(detection_save_path)
baseline = BaselineModel(model, 600, trials=10) baseline = BaselineModel(model, 600, trials=10)
bf.append(baseline.get_baseline_frequency()) bf.append(baseline.get_baseline_frequency())
vs.append(baseline.get_vector_strength()) vs.append(baseline.get_vector_strength())