1 files changed, 5 insertions, 1 deletions
diff --git a/lib/functions.py b/lib/functions.py
index 3630215..399072b 100644
--- a/lib/functions.py
+++ b/lib/functions.py
@@ -470,7 +470,11 @@ class CARTFunction(SKLearnRegressionFunction):
 
 
 class LMTFunction(SKLearnRegressionFunction):
-    pass
+    def get_number_of_nodes(self):
+        return self.regressor.node_count
+
+    def get_max_depth(self):
+        return self.regressor.max_depth
 
 
 class XGBoostFunction(SKLearnRegressionFunction):