diff options
Diffstat (limited to 'lib/parameters.py')
| -rw-r--r-- | lib/parameters.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/lib/parameters.py b/lib/parameters.py index 9cfe145..38e36b2 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -652,7 +652,8 @@ class ModelAttribute: 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() + ret["decision tree/nodes"] = self.model_function.get_number_of_nodes() + ret["decision tree/max depth"] = self.model_function.get_max_depth() return ret |