diff options
Diffstat (limited to 'lib/model.py')
| -rw-r--r-- | lib/model.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/lib/model.py b/lib/model.py index 972547d..6718090 100644 --- a/lib/model.py +++ b/lib/model.py @@ -304,6 +304,8 @@ class AnalyticModel: self.attr_by_name[name][attr].build_cart() elif model_type == "decart": self.attr_by_name[name][attr].build_decart() + elif model_type == "fol": + self.attr_by_name[name][attr].build_fol() elif model_type == "lmt": self.attr_by_name[name][attr].build_lmt() elif model_type == "xgb": @@ -332,7 +334,6 @@ class AnalyticModel: else: paramfit = ParamFit() tree_allowed = bool(int(os.getenv("DFATOOL_DTREE_ENABLED", "1"))) - use_fol = bool(int(os.getenv("DFATOOL_FIT_FOL", "0"))) use_symreg = bool(int(os.getenv("DFATOOL_FIT_SYMREG", "0"))) tree_required = dict() @@ -341,8 +342,6 @@ class AnalyticModel: for attr in self.attr_by_name[name].keys(): if self.attr_by_name[name][attr].function_override is not None: self.attr_by_name[name][attr].fit_override_function() - elif use_fol: - self.attr_by_name[name][attr].build_fol_model() elif use_symreg: self.attr_by_name[name][attr].build_symreg_model() elif self.attr_by_name[name][ |