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.