change base function to line

This commit is contained in:
a.ott 2020-01-20 16:20:40 +01:00
parent 61605cbb9e
commit 4928fedc4e

View File

@ -1,12 +1,12 @@
from stimuli.AbstractStimulus import AbstractStimulus
from models.AbstractModel import AbstractModel
import numpy as np
from functions import full_boltzmann
from functions import line
class FirerateModel(AbstractModel):
KEYS = ["function_params", "f_zero", "adaptation_factor", "a_zero", "a_tau", "step_size"]
VALUES = [[200, 0, 0.5, 0], -np.inf, 0.05, 0, 8, 0.01]
VALUES = [[25, 1], -np.inf, 0.05, 0, 8, 0.01]
def __init__(self, params: dict = None):
super().__init__(params)
@ -49,7 +49,7 @@ class FirerateModel(AbstractModel):
def frequency_step(self, current_stimulus):
params = self.parameters["function_params"]
return full_boltzmann(current_stimulus, params[0], params[1], params[2], params[3])
return line(current_stimulus, params[0], params[1])
def simulates_voltage_trace(self) -> bool:
return False