Skip to content
Snippets Groups Projects

clean scripts

Merged Charly Lamothe requested to merge wip_clean_scripts into master
4 files
+ 30
30
Compare changes
  • Side-by-side
  • Inline

Files

@@ -9,7 +9,7 @@ import numpy as np
class OmpForestRegressor(BaseEstimator):
default_score_metric = 'mse'
DEFAULT_SCORE_METRIC = 'mse'
def __init__(self, models_parameters):
self._regressor = RandomForestRegressor(n_estimators=models_parameters.forest_size,
@@ -60,7 +60,7 @@ class OmpForestRegressor(BaseEstimator):
return predictions
def score(self, X, y, metric=default_score_metric):
def score(self, X, y, metric=DEFAULT_SCORE_METRIC):
"""
Evaluate OMPForestRegressor on (`X`, `y`) using `metric`
Loading