diff options
Diffstat (limited to 'lib/parameters.py')
| -rw-r--r-- | lib/parameters.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/lib/parameters.py b/lib/parameters.py index 5ebf25c..40d0ba5 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -572,6 +572,10 @@ class ModelAttribute: def to_dref(self, unit=None): ret = {"mean": (self.mean, unit), "median": (self.median, unit)} + + if type(self.model_function) is df.SplitFunction: + ret["decision tree nodes"] = self.model_function.get_number_of_nodes() + return ret def webconf_function_map(self): |