Attend the Women in ML Symposium on December 7 Register now

Module: tfrs.layers.factorized_top_k

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.