Train an Estimator
Train an estimator on a set of input data provides by the input_fn().
# S3 method for tf_estimator
train(object, input_fn, steps = NULL, hooks = NULL,
max_steps = NULL, saving_listeners = NULL, ...)Arguments
| object | A TensorFlow estimator. |
| input_fn | An input function, typically generated by the |
| steps | The number of steps for which the model should be trained on
this particular |
| hooks | A list of R functions, to be used as callbacks inside the
training loop. By default, |
| max_steps | The total number of steps for which the model should be
trained. If set, |
| saving_listeners | (Available since TensorFlow v1.4) A list of |
| ... | Optional arguments, passed on to the estimator's |
Value
A data.frame of the training loss history.
See also
Other custom estimator methods: estimator_spec,
estimator,
evaluate.tf_estimator,
export_savedmodel.tf_estimator,
predict.tf_estimator