Merge pull request #125 from EwoutH/sklearn-depreciation

feature_scoring: fix Regressor criterion depreciation
quaquel · May 16, 2022 · c1da7cf · c1da7cf
2 parents a99b71b + 6a443d5
commit c1da7cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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")