build_model: build_model

Description Usage Arguments Value Author(s)

View source: R/build_model.R

Description

define and compile keras model

Usage

1
build_model(units, activation, input_shape, loss, optimizer)

Arguments

units

vector list with number of neurons per layer (last layer has to be 1)

activation

Activation function (either native from keras or custom)

input_shape

number of neurons in the input layer, read from the pre processed dataset

loss

loss function name (defaul: mean_absolute_error)

optimizer

Keras optimizer

Value

Untrained model

Author(s)

Marcos Alves


mppalves/mrscpmule documentation built on Sept. 2, 2021, 9:50 p.m.