j# ddlZddlmZddlmZddlmZdZn #e$rdZYnwxYwddl m Z ddl m Z m Z d Zd Zd Zd ZGd de ZGddeZeZGddeZGddeZGddeZdS)N)SegLST) CPErrorRate) OrcErrorRateTF) BaseMetric)DetailsMetricComponentstotal insertiondeletion substitutioncNeZdZdededefdZedefdZde de fdZ dS) BaseWordErrorRate reference hypothesisreturnctd|D}t|dkrtdt|td|D}t|dkrtdt|||kr/tdt|dt||S)aValidate that both reference and hypothesis are single-session Parameters ---------- reference : SegLST Reference transcription segments. hypothesis : SegLST Hypothesis transcription segments. Returns ------- session_id : str The single session ID found in both reference and hypothesis. c3&K|] }|dV dS session_idN.0ss /home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib/python3.11/site-packages/pyannote/metrics/transcription.py z-BaseWordErrorRate.validate..As&#G#GAlO#G#G#G#G#G#Gz.Is&$I$IQ|_$I$I$I$I$I$Irz=Hypothesis must contain exactly one session. Found sessions: zZReference and hypothesis must refer to the same single session. Found reference sessions: z and hypothesis sessions: )setlen ValueErrorlistpop)selfrrreference_session_idshypothesis_session_idss rvalidatezBaseWordErrorRate.validate0s2"!$#G#GY#G#G#G G G $ % % * *A#'(=#>#>AA  "%$I$Ij$I$I$I!I!I % & &! + +B#'(>#?#?BB  !$: : :K-12G-H-HKK,01G,H,HKK  %((***rc6ttttgS)z%Return the list of metric components.)TOTAL INSERTIONDELETION SUBSTITUTIONclss rmetric_componentsz#BaseWordErrorRate.metric_componentsZs       rdetailc|t|tz|tz}|t}|dkr |dkrdSdS||z S)Ngrg?)r*r,r+r))r$r0 numerator denominators rcompute_metricz BaseWordErrorRate.compute_metricdsR9%|(<.7  AgJ      rc|dSN start_timerrs rz2WordErrorRate.compute_components..s ,r)keyc3&K|] }|dV dSrNrrs rrz3WordErrorRate.compute_components..rPrc|dSrRrrTs rrUz2WordErrorRate.compute_components..s <rrO)r'joinsortedmeetevalwersiso_word_error_rater)lengthr* insertionsr+ deletionsr, substitutions)r$rr_ref_txthyp_txtresults rcompute_componentsz WordErrorRate.compute_componentss MM)Z 0 0((  &y6O6O P P P     ((  &z7P7P Q Q Q     22w  GW#5"6   6= v( f& &.   r r5r6r7__doc__r9r8r@rFrrrf __classcell__rIs@rr<r<psC[#####             rr<cReZdZdZedefdZfdZdedede fdZ xZ S)+ConcatenatedMinimumPermutationWordErrorRatez9Concatenated minimum-Permutation Word Error Rate (cpWER).rcdS)r>cpWERrr-s rr@z7ConcatenatedMinimumPermutationWordErrorRate.metric_names wrc ftstdtjdi|dS)NzConcatenatedMinimumPermutationWordErrorRate metric is not available. Install `pyannote.metrics` with the `transcription` extra to use it.rrBrGs rrFz4ConcatenatedMinimumPermutationWordErrorRate.__init__rJrrrc|||}tj|||}t|jt |jt|j t|j iS)N) r'r[r\cpwerr)r^r*r_r+r`r,rar$rrrres rrfz>ConcatenatedMinimumPermutationWordErrorRate.compute_componentssk ]]9j99 'l00      6= v( f& &.   rrgrjs@rrlrlsCCC[#####             rrlcZeZdZdZedefdZd deffd Zde de de fd Z xZ S) .TimeConstrainedMinimumPermutationWordErrorRatezTime-Constrained minimum-Permutation Word Error Rate (tcpWER). Parameters ---------- collar : float, optional Collar applied to hypothesis pseudo-word level timings, in seconds. Defaults to 5 seconds. rcdS)r>tcpWERrr-s rr@z:TimeConstrainedMinimumPermutationWordErrorRate.metric_names xr@collarc ttstdtjdi|||_dS)NzTimeConstrainedMinimumPermutationWordErrorRate metric is not available. Install `pyannote.metrics` with the `transcription` extra to use it.rrCrDrErFrxr$rxrHrIs rrFz7TimeConstrainedMinimumPermutationWordErrorRate.__init__M$ W  ""6""" rrrc|||}tj|||j|}t |jt|jt|j t|j iSN)rx) r'r[r\tcpwerrxr)r^r*r_r+r`r,rarrs rrfzATimeConstrainedMinimumPermutationWordErrorRate.compute_componentssq ]]9j99 'l11 z$+2   6= v( f& &.   rrw r5r6r7rhr9r8r@r:rFrrrfrirjs@rrtrtsC[u             rrtcZeZdZdZedefdZd deffd Zde de de fd Z xZ S) 7TimeConstrainedOptimalReferenceCombinationWordErrorRatezTime-Constrained Optimal Reference Combination Word Error Rate (tcorcWER) Parameters ---------- collar : float, optional Collar applied to hypothesis pseudo-word level timings, in seconds. Defaults to 5 seconds. rcdS)NtcorcWERrr-s rr@zCTimeConstrainedOptimalReferenceCombinationWordErrorRate.metric_nameszrrwrxc ttstdtjdi|||_dS)NzTimeConstrainedOptimalReferenceCombinationWordErrorRate metric is not available. Install `pyannote.metrics` with the `transcription` extra to use it.rrzr{s rrFz@TimeConstrainedOptimalReferenceCombinationWordErrorRate.__init__r|rrrc|||}tj|||j|}t |jt|jt|j t|j iSr~) r'r[r\tcorcwerrxr)r^r*r_r+r`r,rarrs rrfzJTimeConstrainedOptimalReferenceCombinationWordErrorRate.compute_components sq ]]9j99  (|44 z$+ 5    6= v( f& &.   rrrrjs@rrrsC[u             rr)r[meeteval.io.seglstrmeeteval.wer.wer.cprmeeteval.wer.wer.orcrrCrDpyannote.metrics.baserpyannote.metrics.typesrrr)r*r+r,rr<r?rlrtrrrrrs6 "OOO))))))//////111111 """!"-,,,,,<<<<<<<<   >+>+>+>+>+ >+>+>+B) ) ) ) ) %) ) ) X$ $ $ $ $ 2C$ $ $ N+ + + + + 5F+ + + \* * * * * >O* * * * * s %%