Skip to content

Instantly share code, notes, and snippets.

@glemaitre
Created May 29, 2018 15:02
Show Gist options
  • Save glemaitre/d2f7e4427d73a90cb8757c2bca35b588 to your computer and use it in GitHub Desktop.
Save glemaitre/d2f7e4427d73a90cb8757c2bca35b588 to your computer and use it in GitHub Desktop.
from sklearn.datasets import load_diabetes
from sklearn.compose import TransformedTargetRegressor
from sklearn.model_selection import GridSearchCV
from sklearn.ensemble import RandomForestRegressor
from sklearn.linear_model import LinearRegression
X, y = load_diabetes(return_X_y=True)
ttr = TransformedTargetRegressor()
grid = [{'regressor': [RandomForestRegressor()], 'regressor__n_estimators': [10, 20]},
{'regressor': [LinearRegression()], 'regressor__fit_intercept': [True, False]}]
reg = GridSearchCV(ttr, grid)
reg.fit(X, y)
reg.best_estimator_
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment