diff options
Diffstat (limited to 'lib/model.py')
| -rw-r--r-- | lib/model.py | 19 |
1 files changed, 18 insertions, 1 deletions
diff --git a/lib/model.py b/lib/model.py index 3f784cc..590c6f0 100644 --- a/lib/model.py +++ b/lib/model.py @@ -5,7 +5,12 @@ import numpy as np import os from .automata import PTA, ModelAttribute from .functions import StaticFunction, SubstateFunction, SplitFunction -from .parameters import ParallelParamStats, ParamStats, codependent_param_dict +from .parameters import ( + ParallelParamStats, + ParamStats, + codependent_param_dict, + distinct_param_values, +) from .paramfit import ParamFit from .utils import soft_cast_int, by_name_to_by_param, regression_measures @@ -118,6 +123,12 @@ class AnalyticModel: if self._num_args is None: self._num_args = _num_args_from_by_name(by_name) + self.distinct_param_values_by_name = dict() + for name in self.names: + self.distinct_param_values_by_name[name] = distinct_param_values( + by_name[name]["param"] + ) + self.fit_done = False if compute_stats: @@ -535,6 +546,12 @@ class PTAModel(AnalyticModel): self.pta = pta self.ignore_trace_indexes = ignore_trace_indexes + self.distinct_param_values_by_name = dict() + for name in self.names: + self.distinct_param_values_by_name[name] = distinct_param_values( + by_name[name]["param"] + ) + self.fit_done = False if traces is not None and pelt is not None: |