Regression metric to be optimized
RF is optimizing MSE but we report in the print the RMSE. Please verify this.
diff --git a/supervised/algorithms/sklearn.py b/supervised/algorithms/sklearn.py
index aaaef0a..1fc72f6 100644
--- a/supervised/algorithms/sklearn.py
+++ b/supervised/algorithms/sklearn.py
@@ -144,5 +144,5 @@ def predict_function(estimator, X):
 class SklearnTreesEnsembleRegressorAlgorithm(SklearnTreesEnsembleClassifierAlgorithm):
     def __init__(self, params):
         super(SklearnTreesEnsembleRegressorAlgorithm, self).__init__(params)
-        self.log_metric = Metric({"name": "rmse"})
+        self.log_metric = Metric({"name": "mse"})
         self.predict_function = predict_function
['supervised/algorithms/sklearn.py']