a  dE @sxddlmZmZmZmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZGdddeZGdddeZ Gddde Z!Gddde Z"GdddeZ#GdddeZ$GdddeZ%GdddeZ&Gdd d eZ'Gd!d"d"eZ(Gd#d$d$eZ)Gd%d&d&eZ*d'S)()AnyLiteralOptionalSequence) BLEUScore) CharErrorRate) CHRFScore)ExtendedEditDistance)MatchErrorRate) Perplexity)SacreBLEUScore)SQuAD)TranslationEditRate) WordErrorRate) WordInfoLost)WordInfoPreserved)_deprecated_root_import_classcs8eZdZdZdeeeeee ddfdd Z Z S) _BLEUScorezWrapper for deprecated import. >>> preds = ['the cat is on the mat'] >>> target = [['there is a cat on the mat', 'a cat is on the mat']] >>> bleu = _BLEUScore() >>> bleu(preds, target) tensor(0.7598) FN)n_gramsmoothweightskwargsreturnc s(tddtjf|||d|dS)Nrtext)rrrrsuper__init__)selfrrrr __class__f/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torchmetrics/text/_deprecated.pyrs z_BLEUScore.__init__)rFN) __name__ __module__ __qualname____doc__intboolrrfloatrr __classcell__r!r!rr"rs  rcs(eZdZdZeddfdd ZZS)_CharErrorRatezWrapper for deprecated import. >>> preds = ["this is the prediction", "there is an other sample"] >>> target = ["this is the reference", "there is another one"] >>> cer = _CharErrorRate() >>> cer(preds, target) tensor(0.3415) Nrrc s tddtjfi|dS)Nrrrrrrr!r"r3s z_CharErrorRate.__init__r#r$r%r&rrr*r!r!rr"r+(s r+c s6eZdZdZd eeeeeeeddfdd ZZ S) _CHRFScorezWrapper for deprecated import. >>> preds = ['the cat is on the mat'] >>> target = [['there is a cat on the mat', 'a cat is on the mat']] >>> chrf = _CHRFScore() >>> chrf(preds, target) tensor(0.8640) @FN) n_char_order n_word_orderbeta lowercase whitespacereturn_sentence_level_scorerrc s.tddtjf||||||d|dS)Nrr)r3r4r5r6r7r8r)rr3r4r5r6r7r8rrr!r"rFs z_CHRFScore.__init__)r0r1r2FFF) r#r$r%r&r'r)r(rrr*r!r!rr"r/;s" r/c s:eZdZdZd edeeeeeed d fd d ZZ S)_ExtendedEditDistanceaWrapper for deprecated import. >>> 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) enFr2333333?皙??)r:jaN)languager8alpharhodeletion insertionrrc s.tddtjf||||||d|dS)Nr r)r?r8r@rArBrCr)rr?r8r@rArBrCrrr!r"rgs z_ExtendedEditDistance.__init__)r:Fr2r;r<r=) r#r$r%r&rr(r)rrr*r!r!rr"r9\s" r9cs(eZdZdZeddfdd ZZS)_MatchErrorRatezWrapper for deprecated import. >>> preds = ["this is the prediction", "there is an other sample"] >>> target = ["this is the reference", "there is another one"] >>> mer = _MatchErrorRate() >>> mer(preds, target) tensor(0.4444) Nr,c s tddtjfi|dS)Nr rrr-rr!r"rs z_MatchErrorRate.__init__r.r!r!rr"rD}s rDcs0eZdZdZdeeeddfdd ZZS) _PerplexityaDWrapper for deprecated import. >>> import torch >>> gen = torch.manual_seed(42) >>> preds = torch.rand(2, 8, 5, generator=gen) >>> target = torch.randint(5, (2, 8), generator=gen) >>> target[0, 6:] = -100 >>> perp = _Perplexity(ignore_index=-100) >>> perp(preds, target) tensor(5.8540) N) ignore_indexrrc s$tddtjfd|i|dS)Nr rrFr)rrFrrr!r"rs z_Perplexity.__init__)N) r#r$r%r&rr'rrr*r!r!rr"rEsrEc s@eZdZdZd eeedeeee e ddfdd Z Z S) _SacreBLEUScorezWrapper for deprecated import. >>> preds = ['the cat is on the mat'] >>> target = [['there is a cat on the mat', 'a cat is on the mat']] >>> sacre_bleu = _SacreBLEUScore() >>> sacre_bleu(preds, target) tensor(0.7598) rF13aN)nonerHzhZintlchar)rrtokenizer6rrrc s,tddtjf|||||d|dS)Nr r)rrrLr6rr)rrrrLr6rrrr!r"rs  z_SacreBLEUScore.__init__)rFrHFN) r#r$r%r&r'r(rrrr)rrr*r!r!rr"rGs  rGcs(eZdZdZeddfdd ZZS)_SQuADaLWrapper for deprecated import. >>> preds = [{"prediction_text": "1976", "id": "56e10a3be3433e1400422b22"}] >>> target = [{"answers": {"answer_start": [97], "text": ["1976"]}, "id": "56e10a3be3433e1400422b22"}] >>> squad = _SQuAD() >>> squad(preds, target) {'exact_match': tensor(100.), 'f1': tensor(100.)} Nr,c s tddtjfi|dS)Nr rrr-rr!r"rs z_SQuAD.__init__r.r!r!rr"rMs rMc s4eZdZdZdeeeeeeddfdd ZZS) _TranslationEditRatezWrapper for deprecated import. >>> preds = ['the cat is on the mat'] >>> target = [['there is a cat on the mat', 'a cat is on the mat']] >>> ter = _TranslationEditRate() >>> ter(preds, target) tensor(0.1538) FTN) normalizeno_punctuationr6 asian_supportr8rrc s,tddtjf|||||d|dS)Nrr)rOrPr6rQr8r)rrOrPr6rQr8rrr!r"rs z_TranslationEditRate.__init__)FFTFF)r#r$r%r&r(rrr*r!r!rr"rNs rNcs(eZdZdZeddfdd ZZS)_WordErrorRatezWrapper for deprecated import. >>> preds = ["this is the prediction", "there is an other sample"] >>> target = ["this is the reference", "there is another one"] >>> wer = _WordErrorRate() >>> wer(preds, target) tensor(0.5000) Nr,c s tddtjfi|dS)Nrrrr-rr!r"rs z_WordErrorRate.__init__r.r!r!rr"rRs rRcs(eZdZdZeddfdd ZZS) _WordInfoLostzWrapper for deprecated import. >>> preds = ["this is the prediction", "there is an other sample"] >>> target = ["this is the reference", "there is another one"] >>> wil = _WordInfoLost() >>> wil(preds, target) tensor(0.6528) Nr,c s tddtjfi|dS)Nrrrr-rr!r"r s z_WordInfoLost.__init__r.r!r!rr"rSs rScs(eZdZdZeddfdd ZZS)_WordInfoPreservedzWrapper for deprecated import. >>> preds = ["this is the prediction", "there is an other sample"] >>> target = ["this is the reference", "there is another one"] >>> wip = WordInfoPreserved() >>> wip(preds, target) tensor(0.3472) Nr,c s tddtjfi|dS)Nrrrr-rr!r"rs z_WordInfoPreserved.__init__r.r!r!rr"rTs rTN)+typingrrrrZtorchmetrics.text.bleurZtorchmetrics.text.cerrZtorchmetrics.text.chrfrZtorchmetrics.text.eedr Ztorchmetrics.text.merr Ztorchmetrics.text.perplexityr Ztorchmetrics.text.sacre_bleur Ztorchmetrics.text.squadr Ztorchmetrics.text.terrZtorchmetrics.text.werrZtorchmetrics.text.wilrZtorchmetrics.text.wiprZtorchmetrics.utilities.printsrrr+r/r9rDrErGrMrNrRrSrTr!r!r!r"s2             !!