diff options
| author | Daniel Friesel <daniel.friesel@uos.de> | 2022-01-28 13:16:46 +0100 |
|---|---|---|
| committer | Daniel Friesel <daniel.friesel@uos.de> | 2022-01-28 13:16:46 +0100 |
| commit | 1f391139530f8051a1ece4fcd9de46afe6afde06 (patch) | |
| tree | b69e3d60b9c07ae252651bc236f22f5bb84224d5 /lib/parameters.py | |
| parent | f886ffd67396c2a512bfd71dc62e8bbc72f1ef8b (diff) | |
PTAModel: Add force_tree option
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() |