search

ckaus / EpiPy

A tool for fitting epidemic models.
MIT License
3 stars 0 forks source link

Different parameter representation #14

Closed ckaus closed 8 years ago

ckaus commented 8 years ago

without parameters: ... Model Parameters: [ 0.08827059 0.03064064] ...

vs.

with parameters: ... Model Parameters: {'beta': 0.08827059, 'gamma': 0.03064064} ...

The problem is in class basemodel.py, the return value of param is inconsistent in lines 22 - 34

ckaus commented 8 years ago

basemodel.py in line 27: param, pcov = optimize.curve_fit(self.fit_model, xdata, ydata) variable param is a numpy.ndarray which is inconsistent with given function variable param (which is a dict). New: Function def fit(...) in basemodel.py returns the parameters as list.