tune_model: Tune neural network hyperparameters using Bayesian...

In c7rishi/hidgenclassifier: Functions for Bayesian hierarchical hidden genome classifier

Tune neural network hyperparameters using Bayesian optimization

Description

Tune neural network hyperparameters using Bayesian optimization

Usage

tune_model(
  X_train_norm,
  Y_train_onehot,
  X_val_norm,
  Y_val_onehot,
  trials = 200,
  epochs = 50,
  verbose_keras = 0,
  verbose_mbo = T,
  seed = NULL,
  batch_size = 128
)

Arguments

trials	Number of trials
verbose_keras	Keras verbosity mode (0 = silent, 1 = progress bar, 2 = one line per epoch)
verbose_mbo	Bayesian optimization verbosity mode (logical)

Value

See description of mbo() function from mlrMBO

c7rishi/hidgenclassifier documentation built on June 14, 2024, 11:10 a.m.