diff options
| author | Daniel Friesel <daniel.friesel@uos.de> | 2022-10-18 12:32:45 +0200 |
|---|---|---|
| committer | Daniel Friesel <daniel.friesel@uos.de> | 2022-10-18 12:32:45 +0200 |
| commit | 0fa46d5aa180031b195c167f2bfa4720e4fc9e28 (patch) | |
| tree | af1857b0089f559b24c67130de77c25e764c7370 /lib/parameters.py | |
| parent | 00bf254e111d1642de68259f68a73bbb0907566b (diff) | |
--export-dref: calculate model complexity
Diffstat (limited to 'lib/parameters.py')
| -rw-r--r-- | lib/parameters.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/lib/parameters.py b/lib/parameters.py index 51c47b8..00260fc 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -642,6 +642,9 @@ class ModelAttribute: def to_dref(self, unit=None): ret = {"mean": (self.mean, unit), "median": (self.median, unit)} + if issubclass(type(self.model_function), df.ModelFunction): + ret["complexity"] = self.model_function.get_complexity_score() + if type(self.model_function) in ( df.SplitFunction, df.CARTFunction, |