您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

如何在scikit-learn中使用管道调整自定义内核函数的参数

如何在scikit-learn中使用管道调整自定义内核函数的参数

使用以下函数包装模型:

def GBC(self):
        model = GradientBoostingRegressor()
        p = [{'learning_rate':[[0.0005,0.01,0.02,0.03]],'n_estimators':[[for i in range(1,100)]],'max_depth':[[4]]}]
        return model,p

然后通过参数网格使用内核对其进行测试:

def kernel(self,model,p):
        parameter = ParameterGrid(p)
        clf = gridsearchcv(model, parameter, cv=5, scoring='neg_mean_squared_error',n_jobs=2)
        clf.fit(X,Y)

使用这种方法,您可以在不同的函数上管理函数的种类及其超参数集,直接在main中调用函数

a = the_class()
a.kernel(a.GBC())
其他 2022/1/1 18:26:47 有240人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶