From dacd6b2b62e346084a84b7d4f73abbd6cf62f262 Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Tue, 1 Feb 2022 07:26:23 +0100 Subject: dref export: static and analytic function ≙ single-node tree MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- lib/parameters.py | 3 +++ 1 file changed, 3 insertions(+) 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 -- cgit v1.2.3