i% ddlmZddlmZmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZGddeZ Gdde Z!Gdde Z"Gdde Z#GddeZ$GddeZ%GddeZ&GddeZ'Gd d!eZ(Gd"d#eZ)Gd$d%eZ*Gd&d'eZ+y()))Sequence)AnyLiteralOptional) BLEUScore) CharErrorRate) CHRFScore)ExtendedEditDistance)MatchErrorRate) Perplexity)SacreBLEUScore)SQuAD)TranslationEditRate) WordErrorRate) WordInfoLost)WordInfoPreserved)_deprecated_root_import_classc LeZdZdZ d dededeeede ddf fd Z xZ 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) Nn_gramsmoothweightskwargsreturnc Dtddt|d|||d|y)Nrtext)rrrrsuper__init__)selfrrrr __class__s r/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/torchmetrics/text/_deprecated.pyr z_BLEUScore.__init__s( &k6: QvwQ&Q)FN) __name__ __module__ __qualname____doc__intboolrrfloatrr __classcell__r"s@r#rrs]-1 RRR(5/* R  R  RRr$rc,eZdZdZdeddffd ZxZS)_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) rrNc <tddt|di|y)Nrrrrr!rr"s r#r z_CharErrorRate.__init__4s &ov> "6"r$r&r'r(r)rr r-r.s@r#r0r0)$## ##r$r0cReZdZdZ d dededededededed d ffd ZxZ 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) n_char_order n_word_orderbeta lowercase whitespacereturn_sentence_level_scorerrNc Jtddt|d||||||d|y)Nr r)r7r8r9r:r;r<rr) r!r7r8r9r:r;r<rr"s r#r z_CHRFScore.__init__Gs= &k6:  %%!(C   r$)@FFF) r&r'r(r)r*r,r+rr r-r.s@r#r6r6<st ,1         &*     r$r6cXeZdZdZ d deddededededed ed d ffd ZxZ 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) language)enjar<alpharhodeletion insertionrrNc Jtddt|d||||||d|y)Nr r)rCr<rFrGrHrIrr) r!rCr<rFrGrHrIrr"s r#r z_ExtendedEditDistance.__init__hs> &&>> 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) rrNc <tddt|di|y)Nr rrrr2s r#r z_MatchErrorRate.__init__s  &&6? "6"r$r3r.s@r#rLrL~r4r$rLc:eZdZdZ ddeededdffd ZxZS) _Perplexitya Wrapper for deprecated import. >>> from torch import rand, randint >>> preds = rand(2, 8, 5) >>> target = randint(5, (2, 8)) >>> target[0, 6:] = -100 >>> perp = _Perplexity(ignore_index=-100) >>> perp(preds, target) tensor(5.8540) N ignore_indexrrc @tddt|dd|i|y)Nr rrPrr)r!rPrr"s r#r z_Perplexity.__init__s$ &lF; =l=f=r$)N) r&r'r(r)rr*rr r-r.s@r#rOrOs7 '+>sm>>  >>r$rOc^eZdZdZ d dedededdedeee d e d dffd Z xZ 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) Nrrtokenize)none13azhintlcharr:rrrc Htddt|d|||||d|y)Nr r)rrrTr:rrr)r!rrrTr:rrr"s r#r z_SacreBLEUScore.__init__s6 &&6?  &8yZa ek r$)r%FrVFN) r&r'r(r)r*r+rrrr,rr r-r.s@r#rSrSsvAF-1      =>    (5/*        r$rSc,eZdZdZdeddffd ZxZS)_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.)} rrNc <tddt|di|y)Nrrrrr2s r#r z_SQuAD.__init__s%gv6 "6"r$r3r.s@r#r\r\ ####r$r\cLeZdZdZ d dededededededd ffd ZxZS) _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) normalizeno_punctuationr: asian_supportr<rrNc Htddt|d|||||d|y)Nrr)rarbr:rcr<rr)r!rarbr:rcr<rr"s r#r z_TranslationEditRate.__init__s; &&;VD  )'(C    r$)FFTFF)r&r'r(r)r+rr r-r.s@r#r`r`sg $#,1        &*      r$r`c,eZdZdZdeddffd ZxZS)_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) rrNc <tddt|di|y)Nrrrrr2s r#r z_WordErrorRate.__init__s%ov> "6"r$r3r.s@r#rfrfr^r$rfc,eZdZdZdeddffd ZxZS) _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) rrNc <tddt|di|y)Nrrrrr2s r#r z_WordInfoLost.__init__ s%nf= "6"r$r3r.s@r#ririr^r$ric,eZdZdZdeddffd ZxZS)_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) rrNc <tddt|di|y)Nrrrrr2s r#r z_WordInfoPreserved.__init__s%&96B "6"r$r3r.s@r#rlrlr^r$rlN),collections.abcrtypingrrrtorchmetrics.text.bleurtorchmetrics.text.cerrtorchmetrics.text.chrfr torchmetrics.text.eedr torchmetrics.text.merr torchmetrics.text.perplexityr torchmetrics.text.sacre_bleur torchmetrics.text.squadrtorchmetrics.text.terrtorchmetrics.text.werrtorchmetrics.text.wilrtorchmetrics.text.wiprtorchmetrics.utilities.printsrrr0r6rBrLrOrSr\r`rfrirlrr$r#r}s$)),/,6037)5/.3GRR,#]#&  B 0 B#n#&>*>, n 4 #U #  . > #] # #L # #* #r$