diff --git a/src/pyversity/pyversity.py b/src/pyversity/pyversity.py index 67cf2a1..75eb23b 100644 --- a/src/pyversity/pyversity.py +++ b/src/pyversity/pyversity.py @@ -10,7 +10,7 @@ def diversify( embeddings: np.ndarray, scores: np.ndarray, k: int, - strategy: Strategy | str = Strategy.MMR, + strategy: Strategy | str = Strategy.DPP, diversity: float = 0.5, **kwargs: Any, ) -> DiversificationResult: @@ -21,7 +21,7 @@ def diversify( :param scores: Scores (relevances) of the items to be diversified. :param k: The number of items to select for the diversified result. :param strategy: The diversification strategy to apply. - Supported strategies are: 'mmr' (default), 'msd', 'cover', 'dpp', and 'ssd'. + Supported strategies are: 'dpp' (default), 'mmr', 'msd', 'cover', and 'ssd'. :param diversity: Diversity parameter (range of [0, 1]). Higher values prioritize diversity and lower values prioritize relevance. :param **kwargs: Additional keyword arguments passed to the specific strategy function. :return: A DiversificationResult containing the selected item indices,