i`$ddlmZddlmZmZddlmZddlmZmZm Z ddlm Z m Z m Z ddl mZmZmZmZmZmZddlmZmZdd lmZeZeZeZeZeZd ZGd d eeZGd deeZ Gddee Z!y))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 de de de d e f fd Z dd e d e d e ede 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 t/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/pyannote/metrics/identification.py metric_namez#IdentificationErrorRate.metric_nameKsc8tttttgSr) IER_TOTAL IER_CORRECTIER_FALSE_ALARMIER_MISS IER_CONFUSIONrs rmetric_componentsz)IdentificationErrorRate.metric_componentsOs   X   r! confusionmiss false_alarmcollar skip_overlapc t|di|t|_||_||_||_||_||_yN) super__init__rmatcher_r)r*r+r,r-)selfr)r*r+r,r-kwargs __class__s rr2z IdentificationErrorRate.__init__WsC "6"$ " & (r! reference hypothesisuemc >|j}| |j}| |j}|j|||||d\}} } | D]} | j} |j | d} | j | d}|j | |\}}|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. Tr9r,r-returns_timelineFunique) init_componentsr,r-uemifyduration get_labelsr3r#r$r'r&r%)r4r7r8r9r,r-r5detailRHcommon_timelinesegmentrArhcounts_s rcompute_componentsz*IdentificationErrorRate.compute_componentsgs6.%%' >[[F  ,,L $ zs !!,!#1o 'G''H WU 3A WU 3A a+IFA 9 F9,=!= =  ; 8f[.A#A A  = !X}0E%E E ! 8 6(+; ; ;  ? #x&2I'I I #!'$ r!rCcd|j|tz|j|tzz|j|t zzz}d|t z}|dk(r|dk(ryy||z S)N?r)r)r'r+r%r*r&r#)r4rC numerator denominators rcompute_metricz&IdentificationErrorRate.compute_metrics| !66  6/#::; F8,,- 6),, " A~{* *r!)rNrNrNrOF)NNN)__name__ __module__ __qualname____doc__ classmethodstrr rr(floatboolr2rrrrrLrR __classcell__r6s@rrr2s0C"2%'!&(!#&+ )!))$) ) $ )&6:59;? 6&06'16!) 26$,E? 6 *2% 6 )0 6p+W++r!rc PeZdZdZd dedeffd Z d dededee de 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,r-c ^t|di|||_||_t |_yr/r1r2r,r-rr3r4r,r-r5r6s rr2z IdentificationPrecision.__init__+ "6" ($ r!r7r8r9rc |j}|j||||j|jd\}}}|D]} | j} |j | d} |j | d} |j | | \} }|txx| t| zz cc<|txx| | tzz cc<|SNTr;Fr=) r?r@r,r-rArBr3r lenr r$r4r7r8r9r5rCrDrErFrGrArHrIrJrKs rrLz*IdentificationPrecision.compute_componentss %%' $ zs;;T->->!!,!#1o 'G''H WU 3A WU 3A a+IFA & '8c!f+< < ' / 06+.. / 0'  r!rOFr rSrTrUrVrYrZr2rrrrrLr[r\s@rr^r^sN 'u''6:&0'1!) 2)0 r!r^c PeZdZdZd dedeffd Z d dededee de 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,r-c ^t|di|||_||_t |_yr/r`ras rr2zIdentificationRecall.__init__rbr!r7r8r9rc |j}|j||||j|jd\}}}|D]} | j} |j | d} |j | d} |j | | \} }|txx| | tzz cc<|txx| | tzz cc<|Srd) r?r@r,r-rArBr3r r#r r$rfs rrLz'IdentificationRecall.compute_componentss %%' $ zs;;T->->!!,!#1o 'G''H WU 3A WU 3A a+IFA ? #x&2C'C C # , -F;rss|8.NNDD..-"    # ! &{+oz{+|0oy0f/?F/r!