RSVP for your your local TensorFlow Everywhere event today!

Module: tfrs.layers.factorized_top_k

Layers for retrieving top K recommendations from factorized retrieval models.

Classes

class BruteForce: Brute force retrieval.

class ScaNN: ScaNN approximate retrieval index for a factorized retrieval model.

class Streaming: Retrieves K highest scoring items and their ids from a large dataset.

class TopK: Interface for top K layers.