define_model: Define a Keras model.

define_modelR Documentation

Define a Keras model.

Description

Define a Keras model for the customer churn data.

Usage

define_model(recipe, units1, units2, act1, act2, act3)

Arguments

recipe

A recipe object from prepare_recipe().

units1

Number of neurons in the first layer.

units2

Number of neurons in the second layer.

act1

Activation function for layer 1.

act2

Activation function for layer 2.

act3

Activation function for layer 3.

Value

A Keras model. Not compiled or run yet.


D-Se/ML documentation built on April 1, 2022, 10:53 p.m.