Function maximalMarginalRelevance

  • This function implements the Maximal Marginal Relevance algorithm to select a set of embeddings that maximizes the diversity and relevance to a query embedding.

    Parameters

    • queryEmbedding: number[] | number[][]

      The query embedding.

    • embeddingList: number[][]

      The list of embeddings to select from.

    • Optional lambda: number = 0.5

      The trade-off parameter between relevance and diversity.

    • Optional k: number = 4

      The maximum number of embeddings to select.

    Returns number[]

    The indexes of the selected embeddings in the embeddingList.

Generated using TypeDoc