Module: tfrs.layers.factorized_top_k

Stay organized with collections Save and categorize content based on your preferences.

Layers for retrieving top K recommendations from factorized retrieval models.


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.