Skip to content

v0.5.2

Choose a tag to compare

@maciejkula maciejkula released this 16 Jul 21:22
· 137 commits to main since this release

Fixed

  • Fixed error in default arguments to tfrs.experimental.models.Ranking
    (#311).
  • Fix TPUEmbedding layer to use named parameters.

Added

  • Added batch_metrics to tfrs.tasks.Retrieval for measuring how good the
    model is at picking out the true candidate for a query from other candidates
    in the batch.
  • Added tfrs.experimental.layers.embedding.PartialTPUEmbedding layer, which
    uses tfrs.layers.embedding.TPUEmbedding for large embedding lookups and
    tf.keras.layers.Embedding for smaller embedding lookups.