tensorflow-recommenders

What it is and what it does

TensorFlow Recommenders is a Keras-based library for building recommender systems on top of TensorFlow. It abstracts away common patterns in recommendation workflows—embedding layers, ranking tasks, retrieval metrics—so you can focus on model architecture rather than boilerplate. The library is designed with a gentle learning curve but retains flexibility for complex models; it handles data preparation, model formulation, training, evaluation, and deployment as an integrated workflow.

The package depends on tensorflow, tf-keras, and absl-py, and is built to work seamlessly within the TensorFlow ecosystem. It provides task abstractions (like Retrieval) and metric classes (like FactorizedTopK) that simplify the definition of loss functions and evaluation criteria for recommendation problems. The library is actively maintained, with recent commits and a permissive Apache 2.0 license.

Use it for

• Build a matrix factorization model to predict user ratings for unseen movies or products.
• Train a two-tower neural network for candidate retrieval in a large-scale recommendation system.
• Evaluate ranking quality using built-in metrics like FactorizedTopK across a full candidate set.
• Combine user and item embeddings with custom Keras layers to model complex interaction patterns.
• Deploy a trained recommender model using TensorFlow's standard serving infrastructure.

Worth the install?

AI-flagged interpretation of the facts on this page. Verify before relying on it.

Install

tensorflow-recommenders on PyPI

Before you install

Low install friction with a pure-Python wheel. The package is actively maintained with recent commits and depends on core TensorFlow ecosystem libraries (tensorflow, tf-keras, absl-py), all of which are well-established.

Requires TensorFlow 2.x to be installed separately; the package does not bundle TensorFlow itself.

License in practice

Apache 2.0 is permissive; you can use, modify, and distribute this package freely in commercial and private projects with minimal restrictions.

Quickstart

pip install tensorflow-recommenders

import tensorflow_recommenders as tfrs
import tensorflow as tf

class Model(tfrs.Model):
  def __init__(self):
    super().__init__()
    self.user_model = tf.keras.layers.Embedding(input_dim=2000, output_dim=64)
    self.item_model = tf.keras.layers.Embedding(input_dim=2000, output_dim=64)
    self.task = tfrs.tasks.Retrieval()
  
  def compute_loss(self, features, training=False):
    user_embeddings = self.user_model(features["user_id"])
    item_embeddings = self.item_model(features["item_id"])
    return self.task(user_embeddings, item_embeddings)

model = Model()
model.compile(optimizer=tf.keras.optimizers.Adagrad(0.5))

Package facts

Evidence: tensorflow_recommenders-0.7.7-py3-none-any.whl

Tags

More Software Development packages

See also implicit · keras · scikit-surprise · tensorflow-model-optimization · recbole · tf-slim · tensorflow-estimator · keras-nightly · tf-keras-nightly · tf-estimator-nightly