build_model: Build Keras model

In DeepLearningCausal: Causal Inference with Super Learner and Deep Neural Networks

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
)

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.

Value

Keras model object

DeepLearningCausal documentation built on Nov. 6, 2025, 5:08 p.m.