diff options
| author | Daniel Friesel <daniel.friesel@uos.de> | 2022-01-27 10:20:17 +0100 |
|---|---|---|
| committer | Daniel Friesel <daniel.friesel@uos.de> | 2022-01-27 10:20:17 +0100 |
| commit | 937bcec1ed1bd379c226aea5eb8ce5ec95264703 (patch) | |
| tree | d9da5b2102481e1916808031ce31d28fbe8980d5 /lib/model.py | |
| parent | e149c6bc24935ff8383471759c8775d3174ec29d (diff) | |
add LMT support via https://github.com/cerlymarco/linear-tree
Diffstat (limited to 'lib/model.py')
| -rw-r--r-- | lib/model.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/lib/model.py b/lib/model.py index 1270cf6..7a28197 100644 --- a/lib/model.py +++ b/lib/model.py @@ -160,6 +160,7 @@ class AnalyticModel: with_sklearn_cart = bool( int(os.getenv("DFATOOL_DTREE_SKLEARN_CART", "0")) ) + with_lmt = bool(int(os.getenv("DFATOOL_DTREE_LMT", "0"))) with_xgboost = bool(int(os.getenv("DFATOOL_USE_XGBOOST", "0"))) loss_ignore_scalar = bool( int(os.getenv("DFATOOL_DTREE_LOSS_IGNORE_SCALAR", "0")) @@ -174,6 +175,7 @@ class AnalyticModel: with_function_leaves=with_function_leaves, with_nonbinary_nodes=with_nonbinary_nodes, with_sklearn_cart=with_sklearn_cart, + with_lmt=with_lmt, with_xgboost=with_xgboost, loss_ignore_scalar=loss_ignore_scalar, ) @@ -326,6 +328,7 @@ class AnalyticModel: with_sklearn_cart = bool( int(os.getenv("DFATOOL_DTREE_SKLEARN_CART", "0")) ) + with_lmt = bool(int(os.getenv("DFATOOL_DTREE_LMT", "0"))) with_xgboost = bool(int(os.getenv("DFATOOL_USE_XGBOOST", "0"))) loss_ignore_scalar = bool( int(os.getenv("DFATOOL_DTREE_LOSS_IGNORE_SCALAR", "0")) @@ -347,6 +350,7 @@ class AnalyticModel: with_function_leaves=with_function_leaves, with_nonbinary_nodes=with_nonbinary_nodes, with_sklearn_cart=with_sklearn_cart, + with_lmt=with_lmt, with_xgboost=with_xgboost, loss_ignore_scalar=loss_ignore_scalar, ) @@ -427,6 +431,7 @@ class AnalyticModel: with_function_leaves=False, with_nonbinary_nodes=True, with_sklearn_cart=False, + with_lmt=False, with_xgboost=False, loss_ignore_scalar=False, ): @@ -450,6 +455,7 @@ class AnalyticModel: with_function_leaves=with_function_leaves, with_nonbinary_nodes=with_nonbinary_nodes, with_sklearn_cart=with_sklearn_cart, + with_lmt=with_lmt, with_xgboost=with_xgboost, loss_ignore_scalar=loss_ignore_scalar, threshold=threshold, |