From 228c7dc09d936db1a6a6d976d5bb99148494eb7c Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Thu, 17 Mar 2022 09:23:16 +0100
Subject: dataref export: LMT: add number of leaves

---
 lib/parameters.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'lib/parameters.py')

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):
-- 
cgit v1.2.3