tfr.keras.model.GAMScorer

Univariate scorer using GAM.

Inherits From: UnivariateScorer, Scorer

tfr.keras.model.GAMScorer(
    **gam_kwargs
)

The scorer implements Neural Generalized Additive Ranking Model, which is an additive ranking model. See the paper for more details.

Example usage:

scorer=GAMScorer(hidden_layer_dims=[16])

**gam_kwargs A dict of keyward arguments for GAM layers. Please see tfr.keras.layers.GAMlayer for specific list of keyword arguments.

Methods

__call__

View source

__call__(
    context_features: tfr.keras.model.TensorDict,
    example_features: tfr.keras.model.TensorDict,
    mask: tf.Tensor
) -> Union[tf.Tensor, tfr.keras.model.TensorDict]

See Scorer.