diff options
| author | Daniel Friesel <daniel.friesel@uos.de> | 2022-03-17 09:23:16 +0100 |
|---|---|---|
| committer | Daniel Friesel <daniel.friesel@uos.de> | 2022-03-17 09:23:16 +0100 |
| commit | 228c7dc09d936db1a6a6d976d5bb99148494eb7c (patch) | |
| tree | 9ca3e8395dcc7f1c73e8ed5210c4a345dcbc9f27 | |
| parent | 2c14cc60d5f10bfdd8670a7409d3f83f2531c230 (diff) | |
dataref export: LMT: add number of leaves
| -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 d9ad4c0..d83ff85 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -693,6 +693,9 @@ class ModelAttribute: ret["decision tree/nodes"] = 1 ret["decision tree/max depth"] = 1 + if type(self.model_function) in (df.LMTFunction,): + ret["decision tree/leaves"] = self.model_function.get_number_of_leaves() + return ret def to_dot(self): |