Merge pull request #125 from EwoutH/sklearn-depreciation

quaquel · web-flow · commit c1da7cf93a56 · 2022-05-16T15:23:50.000+02:00
feature_scoring: fix Regressor criterion depreciation
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")
 
