Join TensorFlow at Google I/O, May 11-12 Register now

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.