o sÒh§ã@stddlmZddlmZmZmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZGdd„deƒZd S) é)ÚEnum)ÚCallableÚListÚUnion)Úndarray)ÚTensoré)Úcos_simÚ dot_scoreÚ euclidean_simÚ manhattan_simÚpairwise_cos_simÚpairwise_dot_scoreÚpairwise_euclidean_simÚpairwise_manhattan_simc @s®eZdZdZdZdZdZdZdZe de e dfde e e efe e efge ffdd „ƒZe de e dfde e e efe e efge ffd d „ƒZe dee fd d „ƒZdS)ÚSimilarityFunctionaž Enum class for supported similarity functions. The following functions are supported: - ``SimilarityFunction.COSINE`` (``"cosine"``): Cosine similarity - ``SimilarityFunction.DOT_PRODUCT`` (``"dot"``, ``dot_product``): Dot product similarity - ``SimilarityFunction.EUCLIDEAN`` (``"euclidean"``): Euclidean distance - ``SimilarityFunction.MANHATTAN`` (``"manhattan"``): Manhattan distance ÚcosineÚdotÚ euclideanÚ manhattanÚsimilarity_functionÚreturncCóTt|ƒ}|tjkr tS|tjkrtS|tjkrtS|tjkr tSt d  |t  ¡¡ƒ‚)aé Converts a similarity function name or enum value to the corresponding similarity function. Args: similarity_function (Union[str, SimilarityFunction]): The name or enum value of the similarity function. Returns: Callable[[Union[Tensor, ndarray], Union[Tensor, ndarray]], Tensor]: The corresponding similarity function. Raises: ValueError: If the provided function is not supported. Example: >>> similarity_fn = SimilarityFunction.to_similarity_fn("cosine") >>> similarity_scores = similarity_fn(embeddings1, embeddings2) >>> similarity_scores tensor([[0.3952, 0.0554], [0.0992, 0.1570]]) úOThe provided function {} is not supported. Use one of the supported values: {}.) rÚCOSINEr Ú DOT_PRODUCTr Ú MANHATTANr Ú EUCLIDEANr Ú ValueErrorÚformatÚpossible_values©r©r"úh/var/www/html/alpaca_bot/venv/lib/python3.10/site-packages/sentence_transformers/similarity_functions.pyÚto_similarity_fn#s    ÿÿz#SimilarityFunction.to_similarity_fncCr)aÈ Converts a similarity function into a pairwise similarity function. The pairwise similarity function returns the diagonal vector from the similarity matrix, i.e. it only computes the similarity(a[i], b[i]) for each i in the range of the input tensors, rather than computing the similarity between all pairs of a and b. Args: similarity_function (Union[str, SimilarityFunction]): The name or enum value of the similarity function. Returns: Callable[[Union[Tensor, ndarray], Union[Tensor, ndarray]], Tensor]: The pairwise similarity function. Raises: ValueError: If the provided similarity function is not supported. Example: >>> pairwise_fn = SimilarityFunction.to_similarity_pairwise_fn("cosine") >>> similarity_scores = pairwise_fn(embeddings1, embeddings2) >>> similarity_scores tensor([0.3952, 0.1570]) r) rrr rrrrrrrrr r!r"r"r#Úto_similarity_pairwise_fnKs    ÿÿz,SimilarityFunction.to_similarity_pairwise_fncCsdd„tDƒS)ad Returns a list of possible values for the SimilarityFunction enum. Returns: list: A list of possible values for the SimilarityFunction enum. Example: >>> possible_values = SimilarityFunction.possible_values() >>> possible_values ['cosine', 'dot', 'euclidean', 'manhattan'] cSsg|]}|j‘qSr")Úvalue)Ú.0Úmr"r"r#Ú ƒsz6SimilarityFunction.possible_values..)rr"r"r"r#r vs z"SimilarityFunction.possible_valuesN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrÚDOTrrÚ staticmethodrÚstrrrrr$r%rr r"r"r"r#rs*  ÿ þ' ÿ þ*rN)ÚenumrÚtypingrrrÚnumpyrÚtorchrÚutilr r r r r rrrrr"r"r"r#Ús   (