mi_train_BP {MantaID}

Train a three layers neural network model.

Description

Train a three layers neural network model.

Usage

mi_train_BP(
  train,
  test,
  cls = "class",
  path2save = NULL,
  batch_size = 128,
  epochs = 64,
  validation_split = 0.3
)

Arguments

train

A dataframe with class column as label.

test

A dataframe with class column as label.

cls

A character.The name of the label column.

path2save

The folder path to store the model and train history.

batch_size

Integer or NULL.Number of samples per gradient update.

epochs

Number of epochs to train the model.

validation_split

Float between 0 and 1. Fraction of the training data to be used as validation data.

Value

A list object th at contains the prediction confusion matrix and the model object.