diff options
Diffstat (limited to 'lib')
| -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 faa2e96..8e17f2b 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -644,6 +644,9 @@ class ModelAttribute: ): ret["decision tree/nodes"] = self.model_function.get_number_of_nodes() ret["decision tree/max depth"] = self.model_function.get_max_depth() + elif type(self.model_function) in (df.StaticFunction, df.AnalyticFunction): + ret["decision tree/nodes"] = 1 + ret["decision tree/max depth"] = 1 return ret |