Usage
new_model_class(
classname,
initialize = NULL,
call = NULL,
train_step = NULL,
predict_step = NULL,
test_step = NULL,
compute_loss = NULL,
compute_metrics = NULL,
...,
public = list(),
private = list(),
inherit = NULL,
parent_env = parent.frame()
)Arguments
- classname
String, the name of the custom class. (Conventionally, CamelCase).
- initialize, call, train_step, predict_step, test_step, compute_loss, compute_metrics
Optional methods that can be overridden.
- ..., public
Additional methods or public members of the custom class.
- private
Named list of R objects (typically, functions) to include in instance private environments.
privatemethods will have all the same symbols in scope as public methods (See section "Symbols in Scope"). Each instance will have it's ownprivateenvironment. Any objects inprivatewill be invisible from the Keras framework and the Python runtime.- inherit
What the custom class will subclass. By default, the base keras class.
- parent_env
The R environment that all class methods will have as a grandparent.