ensemble.py 文件源码

python
阅读 30 收藏 0 点赞 0 评论 0

项目:eezzy 作者: 3Blades 项目源码 文件源码
def fit(self, df, y, param_grid=None):
        from sklearn.model_selection import GridSearchCV
        X = df.drop(y, axis=1).values
        y = df[y].values

        meta_X = self.get_meta(X)

        if param_grid is not None:
            model = self.stacked_model_class()
            gridsearch = GridSearchCV(model, param_grid)
            gridsearch.fit(meta_X, y)
            self.stacked_model = self.stacked_model_class(**gridsearch.best_params_)
        else:
            self.stacked_model = self.stacked_model_class()

        self.stacked_model.fit(meta_X, y)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号