DeepLearningCausal (version 0.0.107)

build_model: Build Keras model

Description

Specify model for deep learning

Usage

build_model(
  hidden.layer,
  input_shape,
  output_units = 1,
  output_activation = "sigmoid",
  hidden_activation = "relu",
  dropout_rate = NULL
)

Value

Keras model object

Arguments

hidden.layer

vector of integers for number of hidden units in each hidden layer

input_shape

integer for number of input features

output_units

integer for number of output units, default is 1

output_activation

string for output layer activation function, default is "sigmoid"

hidden_activation

string for hidden layer activation function, default is "relu"

dropout_rate

double or vector for proportion of hidden layer to drop out.