如何在GradientBoostingRegressor - BaseEstimator中使用scikit learn？

from sklearn.ensemble import GradientBoostingRegressor
from sklearn.base import BaseEstimator

class MyRegressor(BaseEstimator):
    def __init__(self, **params):
        self.model = GradientBoostingRegressor(**params)

    def fit(self, X, y):
        self.model.fit(X, y)
        return self

    def predict(self, X):
        return self.model.predict(X)

my_regressor = MyRegressor(n_estimators=100, learning_rate=0.1)

my_regressor.fit(X_train, y_train)
predictions = my_regressor.predict(X_test)