diff options
Diffstat (limited to 'lib/parameters.py')
| -rw-r--r-- | lib/parameters.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/lib/parameters.py b/lib/parameters.py index 51ff680..1f72700 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -670,6 +670,7 @@ class ModelAttribute: df.SplitFunction, df.CARTFunction, df.XGBoostFunction, + df.LMTFunction, ): ret["decision tree/nodes"] = self.model_function.get_number_of_nodes() ret["decision tree/max depth"] = self.model_function.get_max_depth() |