sklearn 0.22 will add a new argument to `mean_squared_error` which will allow it to return RMSE. We should include this.