In mlr-org/mlr3keras: mlr3 Keras extension

devtools::load_all()

task = tsk("boston_housing")
lrn = lrn("regr.smlp2")
lrn$train(task)

fp = tempfile(fileext = ".h5")
lrn$save(fp)
system(paste0("netron ", fp))

l = lrn("classif.smlp2")
l$param_set$values$epochs = 300
l$param_set$values$n_max = 24
l$param_set$values$n_layers = 2
l$param_set$values$validation_split = 0.1
l$train(tsk("iris"))
l$plot()
l$model
lrn$model

mlr-org/mlr3keras documentation built on April 12, 2022, 11:35 a.m.