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