diff --git a/ema_workbench/analysis/feature_scoring.py b/ema_workbench/analysis/feature_scoring.py index 456f0016a..052ddbd01 100644 --- a/ema_workbench/analysis/feature_scoring.py +++ b/ema_workbench/analysis/feature_scoring.py @@ -210,7 +210,7 @@ def get_rf_feature_scores( criterion = "gini" elif mode == RuleInductionType.REGRESSION: rfc = RandomForestRegressor - criterion = "mse" + criterion = "squared_error" else: raise ValueError(f"{mode} not valid for mode") @@ -316,7 +316,7 @@ def get_ex_feature_scores( criterion = "gini" elif mode == RuleInductionType.REGRESSION: etc = ExtraTreesRegressor - criterion = "mse" + criterion = "squared_error" else: raise ValueError(f"{mode} not valid for mode")