diff options
Diffstat (limited to 'lib/model.py')
-rw-r--r-- | lib/model.py | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/lib/model.py b/lib/model.py index 5979c70..be63a8a 100644 --- a/lib/model.py +++ b/lib/model.py @@ -311,9 +311,20 @@ class AnalyticModel: return {"by_name": detailed_results} - def to_json(self): - # TODO - pass + def to_json(self) -> dict: + """ + Return JSON encoding of this AnalyticModel. + """ + ret = { + "parameters": self.parameters, + "name": dict([[name, dict()] for name in self.names]), + } + + for name in self.names: + for attr_name, attr in self.attr_by_name[name].items(): + ret["name"][name][attr_name] = attr.to_json() + + return ret def predict(self, trace, with_fitted=True, wth_lut=False): pass |