Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge

Module: tfr.keras.model

Ranking model utilities and classes in tfr.keras.


class AbstractModelBuilder: Interface to build a tf.keras.Model for ranking.

class DNNScorer: Univariate scorer using DNN.

class FeatureSpecInputCreator: InputCreator with feature specs.

class GAMScorer: Univariate scorer using GAM.

class InputCreator: Interface for input creator.

class ModelBuilder: Builds a tf.keras.Model.

class ModelBuilderWithMask: Interface to build a tf.keras.Model for ranking with a mask Tensor.

class Preprocessor: Interface for feature preprocessing.

class PreprocessorWithSpec: Preprocessing inputs with provided spec.

class Scorer: Interface for scorer.

class TypeSpecInputCreator: InputCreator with tensor type specs.

class UnivariateScorer: Interface for univariate scorer.


create_keras_model(...): Creates a Functional Keras ranking model.

Type Aliases

TensorDict: The central part of internal API.

TensorLike: Union of all types that can be converted to a tf.Tensor by tf.convert_to_tensor.