Stay organized with collections Save and categorize content based on your preferences.

Hyperparameters used in ModelFitPipeline.

Hyperparameters to be specified for ranking pipeline.

model_dir A path to output the model and training data.
num_epochs An integer to specify the number of epochs of training.
steps_per_epoch An integer to specify the number of steps per epoch. When it is None, going over the training data once is counted as an epoch.
validation_steps An integer to specify the number of validation steps in each epoch. Note that a mini-batch of data will be evaluated in each step and this is the number of steps taken for validation in each epoch.
learning_rate A float to indicate the learning rate of the optimizer.
loss A string or a map to strings that indicate the loss to be used. When loss is a string, all outputs and labels will be trained with the same loss. When loss is a map, outputs and labels will be trained with losses implied by the corresponding keys.
loss_reduction An option in tf.keras.losses.Reduction to specify the reduction method.
optimizer An option in tf.keras.optimizers identifiers to specify the optimizer to be used.
loss_weights None or a float or a map to floats that indicate the relative weights for each loss. When not specified, all losses are applied with the same weight 1.
steps_per_execution An integer to specify the number of steps executed in each operation. Tuning this to optimize the training performance in distributed training.
automatic_reduce_lr A boolean to indicate whether to use ReduceLROnPlateau callback.
use_weighted_metrics A boolean to indicate whether to use weighted metrics.
export_best_model A boolean to indicate whether to export the best model evaluated by the best_exporter_metric on the validation data.
best_exporter_metric_higher_better A boolean to indicate whether the best_exporter_metric is the higher the better.
best_exporter_metric A string to specify the metric used to monitor the training and to export the best model. Default to the 'loss'.
strategy An option of strategies supported in strategy_utils. Choose from ["MirroredStrategy", "MultiWorkerMirroredStrategy", "ParameterServerStrategy", "TPUStrategy"].
cluster_resolver A cluster_resolver to build strategy.
variable_partitioner Variable partitioner to be used in ParameterServerStrategy.
tpu TPU address for TPUStrategy. Not used for other strategy.



automatic_reduce_lr False
best_exporter_metric 'loss'
best_exporter_metric_higher_better False
cluster_resolver None
export_best_model False
loss_reduction 'auto'
loss_weights None
optimizer 'adam'
steps_per_execution 10
strategy None
tpu ''
use_weighted_metrics False
variable_partitioner None