You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.