View source on GitHub
|
Ranking model utilities and classes in tfr.keras.
Classes
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.
Functions
create_keras_model(...): Creates a Functional Keras ranking model.