首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >R和H2O中的参数优化

R和H2O中的参数优化
EN

Stack Overflow用户
提问于 2016-10-12 18:56:10
回答 2查看 318关注 0票数 2

我需要对RH2o上的gbm模型进行参数优化。我对H2o比较陌生,我认为在执行下面的操作之前,我需要将ntree和learn_rate(下面)转换为H2o向量。我该怎么做这个手术?谢谢!

代码语言:javascript
运行
复制
ntrees <- c(100,200,300,400)
learn_rate <- c(1,0.5,0.1)
for (i in ntrees){
  for j in learn_rate{
    n = ntrees[i]
    l= learn_rate[j]
    gbm_model <- h2o.gbm(features, label, training_frame = train, validation_frame = valid, ntrees=ntrees[[i]],max_depth = 5,learn_rate=learn_rate[j])
    print(c(ntrees[i],learn_rate[j],h2o.mse(h2o.performance(gbm_model, valid = TRUE))))

  }
}
EN

Stack Overflow用户

回答已采纳

发布于 2016-10-12 23:30:09

您可以使用h2o.grid()进行网格搜索。

代码语言:javascript
运行
复制
# specify your hyper parameters
hyper_params = list( ntrees = c(100,200,300,400), learn_rate =  c(1,0.5,0.1) )

# then build your grid
grid <- h2o.grid(
## hyper parameters
hyper_params = hyper_params,

## which algorithm to run
algorithm = "gbm",

## identifier for the grid, to later retrieve it
grid_id = "my_grid", 

## standard model parameters
x = features, 
y = label, 
training_frame = train, 
validation_frame = valid,

## set a seed for reproducibility
seed = 1234)

您可以阅读更多关于h2o.grid()如何在R文档package.pdf中工作的内容。

票数 2
EN
查看全部 2 条回答
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/40006311

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档