AUTO_VI$..init.. {autovi}

Initialization method

Description

This function will be called after an instance is built. User input will be stored in the environment.

Usage

AUTO_VI$..init..(fitted_model, keras_model = NULL, data = NULL, node_index = 1L)

Arguments

fitted_model

Model. A model object, e.g. lm.

keras_model

Keras model. A trained computer vision model.

data

Data frame. The data used to fit the model.

node_index

Integer. An index indicating which node of the output layer contains the visual signal strength. This is particularly useful when the keras model has more than one output nodes.

Value

Return the object itself.

Examples

my_vi <- auto_vi(fitted_model = lm(speed ~ dist, data = cars))
my_vi