scikit-learn:从管道中检索模型对象

时间:2018-08-31 22:17:30

标签: python scikit-learn

我有以下管道构建,我想要做的是获取在管道内部构建的随机森林模型对象。 rf是唯一的初始化,并且没有rf.estimators_

     grid_params = [{'bootstrap': [True],
      'min_samples_leaf': [5],
      'n_estimators': [1000],
      'max_features': [8],
      'min_samples_split': [12],
      'max_depth': [60]}]

     rf = RandomForestRegressor()
     grid_search = GridSearchCV(estimator=rf, param_grid=grid_params)

     pipe = PipelineRF(
         [
             ('grid', grid_search)
         ]
     )
     _ = StratifiedKFold(random_state=125)

有没有办法打电话给pipe.get_model()

0 个答案:

没有答案