GridSearch over MultiOutputRegressor?

前端 未结 3 1780
小鲜肉
小鲜肉 2021-02-01 06:31

Let\'s consider a multivariate regression problem (2 response variables: Latitude and Longitude). Currently, a few machine learning model implementations like Support Vector Reg

3条回答
  •  梦谈多话
    2021-02-01 07:07

    For use without pipeline, put estimator__ before parameters:

    param_grid = {'estimator__min_samples_split':[10, 50],
                  'estimator__min_samples_leaf':[50, 150]}
    
    gb = GradientBoostingRegressor()
    gs = GridSearchCV(MultiOutputRegressor(gb), param_grid=param_grid)
    
    gs.fit(X,y)
    

提交回复
热议问题