summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDaniel Friesel <daniel.friesel@uos.de>2022-03-10 18:23:47 +0100
committerDaniel Friesel <daniel.friesel@uos.de>2022-03-10 18:23:47 +0100
commit49aced3b7b26c605e5be5f1382df49f05a39a452 (patch)
tree556640443ffe94bee4c5598a1266285f47436443
parentfe7a702e6a7724ba8ef31697f5bf60eb800102b3 (diff)
LMT max_depth is broken
-rw-r--r--lib/functions.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/lib/functions.py b/lib/functions.py
index ac92af9..5aa773b 100644
--- a/lib/functions.py
+++ b/lib/functions.py
@@ -539,6 +539,7 @@ class LMTFunction(SKLearnRegressionFunction):
return self.regressor.node_count
def get_max_depth(self):
+ # FIXME this returns the configured maximum depth, not the actual tree depth.
return self.regressor.max_depth