Search code examples
rh2omlr

Set seed for h2o via mlr in R

Is it possible to set a seed for h2o models via mlr? I could only find how to do it in h2o directly, e.g.

gbm_w_seed_2 <- h2o.gbm(x = predictors, y = response, training_frame = train,
                        validation_frame = valid, col_sample_rate =.7 ,
                        seed = 1234)
Solution

  • Yes, these are exposed as learner parameters. For example:

    lrn = makeLearner("classif.h2ogbm", par.vals = list(seed = 123))