- Fixed the bug in
tfrs.layers.loss.SamplingProbablityCorrection
that logits should substract the log of item probability. tfrs.experimental.models.RankingModel
can be used as DLRM like model with Dot Product feature interaction or DCN like model with Cross layer.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
andDotInteraction
layers have been moved totfrs.layers.feature_interaction
package.
TopK
layers now come with aquery_with_exclusions
method, allowing certain candidates to be excluded from top-k retrieval.TPUEmbedding
Keras layer for accelerating embedding lookups for large tables with TPU.
-
factorized_top_k.Streaming
layer now accepts a query model, like otherfactorized_top_k
layers. -
Updated ScaNN to 1.2.0, which requires TensorFlow 2.4.x. When not using ScaNN, any TF >= 2.3 is still supported.
- Pinned TensorFlow to >= 2.3 when ScaNN is not being installed. When ScaNN is being installed, we pin on >= 2.3, < 2.4. This allows users to use TFRS on TF 2.4 when they are not using ScaNN.
- Pinned TensorFlow to 2.3.x and ScaNN to 1.1.1 to ensure TF and ScaNN versions are in lockstep.
- Deep cross networks: efficient ways of learning feature interactions.
- ScaNN integration: efficient approximate maximum inner product search for fast retrieval.
tfrs.tasks.Ranking.call
now accepts acompute_metrics
argument to allow switching off metric computation.tfrs.tasks.Ranking
now accepts label and prediction metrics.- Add metrics setter/getters on
tfrs.tasks.Retrieval
.
-
Corpus retrieval metrics and layers have been reworked.
tfrs.layers.corpus.DatasetTopk
has been removed,tfrs.layers.corpus.DatasetIndexedTopK
renamed totfrs.layers.factorized_top_k.Streaming
,tfrs.layers.ann.BruteForce
renamed totfrs.layers.factorized_top_k.BruteForce
. All top-k retrieval layers (BruteForce
,Streaming
) now follow a common interface.
Dataset
parallelism enabled by default inDatasetTopK
andDatasetIndexedTopK
layers, bringing over 2x speed-ups to evaluations workloads.evaluate_metrics
argument totfrs.tasks.Retrieval.call
renamed tocompute_metrics
.