o yi @shddlmZmZmZmZmZddlmZmZddl m Z ddl m Z m Z ddlmZGdddeZdS) )AnyListSequenceTupleUnion)Tensorstack)Literal) _eed_compute _eed_update)MetriccseZdZUdZdZeed<dZeed<dZeed<e e ed<    dd e d dede de de de de ffdd ZdeeeefdeeeeefddfddZdee ee e fffddZZS)ExtendedEditDistanceaComputes extended edit distance score (`ExtendedEditDistance`_) for strings or list of strings. The metric utilises the Levenshtein distance and extends it by adding a jump operation. As input to ``forward`` and ``update`` the metric accepts the following input: - ``preds`` (:class:`~Sequence`): An iterable of hypothesis corpus - ``target`` (:class:`~Sequence`): An iterable of iterables of reference corpus As output of ``forward`` and ``compute`` the metric returns the following output: - ``eed`` (:class:`~torch.Tensor`): A tensor with the extended edit distance score Args: language: Language used in sentences. Only supports English (en) and Japanese (ja) for now. return_sentence_level_score: An indication of whether sentence-level EED score is to be returned alpha: optimal jump penalty, penalty for jumps between characters rho: coverage cost, penalty for repetition of characters deletion: penalty for deletion of character insertion: penalty for insertion or substitution of character kwargs: Additional keyword arguments, see :ref:`Metric kwargs` for more info. Example: >>> from torchmetrics import ExtendedEditDistance >>> preds = ["this is the prediction", "here is an other sample"] >>> target = ["this is the reference", "here is another one"] >>> eed = ExtendedEditDistance() >>> eed(preds=preds, target=target) tensor(0.3078) Fhigher_is_betteris_differentiablefull_state_update sentence_eeden@333333?皙??languagerjareturn_sentence_level_scorealpharhodeletion insertionkwargsc  stjd i||dvrtd|||_||_tgd||||gD]\}} t| tr7t| tr?| dkr?td|dq%||_||_ ||_ ||_ |j dgdd dS) Nrz?Expected argument `language` to either be `en` or `ja` but got )rrrrrz Parameter `z)` is expected to be a non-negative float.rcat)dist_reduce_fx) super__init__ ValueErrorrrzip isinstancefloatrrrr add_state) selfrrrrrrr param_nameparam __class__r"I/home/ubuntu/.local/lib/python3.10/site-packages/torchmetrics/text/eed.pyr$>s zExtendedEditDistance.__init__predstargetreturnNc Cs(t|||j|j|j|j|j|j|_dS)z*Update state with predictions and targets.N)r rrrrrr)r*r0r1r"r"r/update[s zExtendedEditDistance.updatecCs"t|j}|jr|t|jfS|S)z'Calculate extended edit distance score.)r rrr)r*averager"r"r/computels zExtendedEditDistance.compute)rFrrrr)__name__ __module__ __qualname____doc__rbool__annotations__rrrrr r(rr$rstrrr3rr5 __classcell__r"r"r-r/r sF      &r N)typingrrrrrtorchrrtyping_extensionsr torchmetrics.functional.text.eedr r torchmetrics.metricr r r"r"r"r/s