ó — óhDãó~•SSKJr SSKJr SSKJr SSKJr SSKJ r SSK J r J r J r JrJrJrJrJr "SS \5rg ) é)Ú annotations)ÚEnum)ÚCallable)Úndarray)ÚTensoré)Úcos_simÚ dot_scoreÚ euclidean_simÚ manhattan_simÚpairwise_cos_simÚpairwise_dot_scoreÚpairwise_euclidean_simÚpairwise_manhattan_simcóx•\rSrSrSrSrSrSrSrSr \ S Sj5r \ S Sj5r \ S S j5r S rg )ÚSimilarityFunctionéa† 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Ú manhattancó0•[U5nU[R:Xa[$U[R:Xa[$U[R :Xa[ $U[R:Xa[$[SUS[R5S35e)ay 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]]) úThe provided function ú4 is not supported. Use one of the supported values: Ú.) rÚCOSINEr Ú DOT_PRODUCTr Ú MANHATTANr Ú EUCLIDEANr Ú ValueErrorÚpossible_values©Úsimilarity_functions Úd/home/james-whalen/.local/lib/python3.13/site-packages/sentence_transformers/similarity_functions.pyÚto_similarity_fnÚ#SimilarityFunction.to_similarity_fn%s¬€ô.1Ð1DÓEÐà Ô"4×";Ñ";Ó ;܈NØ Ô"4×"@Ñ"@Ó @ÜÐ Ø Ô"4×">Ñ">Ó >Ü Ð Ø Ô"4×">Ñ">Ó >Ü Ð äØ$Ð%8Ð$9Ð9môoA÷oQñoQóoSðnTðTUð Vó ð ócó0•[U5nU[R:Xa[$U[R:Xa[$U[R :Xa[ $U[R:Xa[$[SUS[R5S35e)aH 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]) rrr) rrr rrrrrrr r!r"s r$Úto_similarity_pairwise_fnÚ,SimilarityFunction.to_similarity_pairwise_fnKs­€ô41Ð1DÓEÐà Ô"4×";Ñ";Ó ;Ü#Ð #Ø Ô"4×"@Ñ"@Ó @Ü%Ð %Ø Ô"4×">Ñ">Ó >Ü)Ð )Ø Ô"4×">Ñ">Ó >Ü)Ð )äØ$Ð%8Ð$9Ð9môoA÷oQñoQóoSðnTðTUð Vó ð r'cóL•[Vs/sHoRPM sn$s snf)a$ 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'] )rÚvalue)Úms r$r!Ú"SimilarityFunction.possible_valuests €õ"4Ó4Ò!3˜A—”Ñ!3Ñ4Ð4ùÒ4s‰!©N)r#zstr | SimilarityFunctionÚreturnz6Callable[[Tensor | ndarray, Tensor | ndarray], Tensor])r0z list[str])Ú__name__Ú __module__Ú __qualname__Ú__firstlineno__Ú__doc__rrÚDOTrrÚ staticmethodr%r)r!Ú__static_attributes__r/r'r$rrs~†ñð€FØ€KØ €CØ€IØ€Iàð# Ø5ð# à ?ó# óð# ðJð& Ø5ð& à ?ó& óð& ðPó 5óó 5r'rN)Ú __future__rÚenumrÚtypingrÚnumpyrÚtorchrÚutilr r r r r rrrrr/r'r$Úr?s1ðÝ"åÝåÝ÷ ÷ ó ôl5˜õl5r'