Module: tfr.keras

TensorFlow Ranking Keras Module.

Modules

layers module: Defines Keras Layers for TF-Ranking.

losses module: Keras losses in TF-Ranking.

metrics module: Keras metrics in TF-Ranking.

model module: Ranking model utilities and classes in tfr.keras.

pipeline module: Ranking pipeline to train tf.keras.Model in tfr.keras.

saved_model module: SavedModel utilities for TF-Ranking.

strategy_utils module: tf.distribute strategy utils for Ranking pipeline in tfr.keras.

utils module: Utils for tfr.keras.