Skip to content

v0.5.0

Compare
Choose a tag to compare
@maciejkula maciejkula released this 07 May 17:47
· 139 commits to main since this release

Added

  • tfrs.experimental.models.Ranking, an experimental pre-built model for
    ranking tasks. Can be used as DLRM like model with Dot Product feature
    interaction or DCN like model with Cross layer.

Changed

  • Fixed the bug in tfrs.layers.loss.SamplingProbablityCorrection that logits
    should subtract the log of item probability.
  • tfrs.experimental.optimizers.CompositeOptimizer: an optimizer that
    composes multiple individual optimizers which can be applied to different
    subsets of the model's variables.
  • tfrs.layers.dcn.Cross and DotInteraction layers have been moved to
    tfrs.layers.feature_interaction package.