j`$ddlmZddlmZmZddlmZddlmZmZm Z ddlm Z m Z m Z ddl mZmZmZmZmZmZddlmZmZdd lmZeZeZeZeZeZd ZGd d eeZGd deeZ Gddee Z!dS))Optional) AnnotationTimeline) BaseMetric) PrecisionPRECISION_RETRIEVEDPRECISION_RELEVANT_RETRIEVED)RecallRECALL_RELEVANTRECALL_RELEVANT_RETRIEVED) LabelMatcher MATCH_TOTAL MATCH_CORRECTMATCH_CONFUSIONMATCH_MISSED_DETECTIONMATCH_FALSE_ALARM)MetricComponentsDetails)UEMSupportMixinzidentification error ratec eZdZdZedefdZedefdZ dde d e d e d e d e f fd Z dde de de ed e e d e e def dZdede fdZxZS)IdentificationErrorRatea[Identification error rate ``ier = (wc x confusion + wf x false_alarm + wm x miss) / total`` where - `confusion` is the total confusion duration in seconds - `false_alarm` is the total hypothesis duration where there are - `miss` is - `total` is the total duration of all tracks - wc, wf and wm are optional weights (default to 1) Parameters ---------- collar : float, optional Duration (in seconds) of collars removed from evaluation around boundaries of reference segments. skip_overlap : bool, optional Set to True to not evaluate overlap regions. Defaults to False (i.e. keep overlap regions). confusion, miss, false_alarm: float, optional Optional weights for confusion, miss and false alarm respectively. Default to 1. (no weight) returnctSN)IER_NAMEclss /home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib/python3.11/site-packages/pyannote/metrics/identification.py metric_namez#IdentificationErrorRate.metric_nameKscBtttttgSr) IER_TOTAL IER_CORRECTIER_FALSE_ALARMIER_MISS IER_CONFUSIONrs rmetric_componentsz)IdentificationErrorRate.metric_componentsOs   X   r!?F confusionmiss false_alarmcollar skip_overlapc tjdi|t|_||_||_||_||_||_dSN) super__init__rmatcher_r+r,r-r.r/)selfr+r,r-r.r/kwargs __class__s rr4z IdentificationErrorRate.__init__WsX ""6"""$ " & (r!N reference hypothesisuemc |}||j}||j}||||||d\}} } | D]} | j} || d} | | d}|| |\}}|txx| |tzz cc<|txx| |tzz cc<|txx| |tzz cc<|txx| |tzz cc<|txx| |tzz cc<|S)ah Parameters ---------- collar : float, optional Override self.collar skip_overlap : bool, optional Override self.skip_overlap See also -------- :class:`pyannote.metric.diarization.DiarizationErrorRate` uses these two options in its `compute_components` method. NTr;r.r/returns_timelineFunique) init_componentsr.r/uemifyduration get_labelsr5r#r$r'r&r%)r6r9r:r;r.r/r7detailRHcommon_timelinesegmentrCrhcounts_s rcompute_componentsz*IdentificationErrorRate.compute_componentsgs.%%'' >[F  ,L $ zs !!,!#!#1o ' J JG'H WU 33A WU 33A a++IFA 9   F9,=!= =    ;   8f[.A#A A    = ! ! !X}0E%E E ! ! ! 8   6(+; ; ;    ? # # #x&2I'I I # # # # r!rEcd|j|tz|j|tzz|j|t zzz}d|t z}|dkr |dkrdSdS||z S)Nr)r*r)r+r'r-r%r,r&r#)r6rE numerator denominators rcompute_metricz&IdentificationErrorRate.compute_metrics~ !66 6/#::; F8,,- 6),, "  A~~rr{* *r!)r)r)r)r*F)NNN)__name__ __module__ __qualname____doc__ classmethodstrr rr(floatboolr4rrrrrNrR __classcell__r8s@rrr2sf0C["2[%'!&(!#&+ ))!))$) ) $ ))))))&6:59;? 66&06'16!) 26$,E? 6 *2% 6 )0 6666p+W+++++++++r!rc VeZdZdZd dedeffd Z dded ed ee d e fd Z xZ S)IdentificationPrecisionaYIdentification Precision Parameters ---------- collar : float, optional Duration (in seconds) of collars removed from evaluation around boundaries of reference segments. skip_overlap : bool, optional Set to True to not evaluate overlap regions. Defaults to False (i.e. keep overlap regions). r*Fr.r/c |tjdi|||_||_t |_dSr1r3r4r.r/rr5r6r.r/r7r8s rr4z IdentificationPrecision.__init__>""6""" ($ r!Nr9r:r;rc |}|||||j|jd\}}}|D]} | j} || d} || d} || | \} }|txx| t| zz cc<|txx| | tzz cc<|SNTr=Fr?) rArBr.r/rCrDr5r lenr r$r6r9r:r;r7rErFrGrHrIrCrJrKrLrMs rrNz*IdentificationPrecision.compute_componentss %%'' $ zs;T->!!,!#!#1o ' / /G'H WU 33A WU 33A a++IFA & ' ' '8c!ff+< < ' ' ' / 0 0 06+.. / 0 0 0 0 r!r*Fr rSrTrUrVrYrZr4rrrrrNr[r\s@rr^r^s  ''u'''''''6:&0'1!) 2)0 r!r^c VeZdZdZd dedeffd Z dded ed ee d e fd Z xZ S)IdentificationRecallaVIdentification Recall Parameters ---------- collar : float, optional Duration (in seconds) of collars removed from evaluation around boundaries of reference segments. skip_overlap : bool, optional Set to True to not evaluate overlap regions. Defaults to False (i.e. keep overlap regions). r*Fr.r/c |tjdi|||_||_t |_dSr1r`ras rr4zIdentificationRecall.__init__rbr!Nr9r:r;rc |}|||||j|jd\}}}|D]} | j} || d} || d} || | \} }|txx| | tzz cc<|txx| | tzz cc<|Srd) rArBr.r/rCrDr5r r#r r$rfs rrNz'IdentificationRecall.compute_componentss %%'' $ zs;T->!!,!#!#1o ' P PG'H WU 33A WU 33A a++IFA ? # # #x&2C'C C # # # , - - -F;rss8........NNNNNNNNNNDDDDDDDDDD................-,,,,,,,""""""    # ! &{+{+{+{+{+oz{+{+{+|00000oy000f/////?F/////r!