diff --git a/models/FirerateModel.py b/models/FirerateModel.py index 3426aa1..d12345c 100644 --- a/models/FirerateModel.py +++ b/models/FirerateModel.py @@ -10,8 +10,8 @@ class FirerateModel(AbstractModel): "f_zero": -np.inf, "adaptation_factor": 0.05, "a_zero": 0, - "a_tau": 8, - "step_size": 0.01} + "a_tau": 0.02, + "step_size": 0.0005} def __init__(self, params: dict = None): super().__init__(params) @@ -32,7 +32,7 @@ class FirerateModel(AbstractModel): freq_trace = [] a_trace = [] - for time_point in np.arange(0, total_time_s * 1000, self.parameters["step_size"]): + for time_point in np.arange(0, total_time_s, self.parameters["step_size"]): freq_trace.append(current_freq) a_trace.append(current_adaptation)