
    bCi0                     v    S SK r SSKJrJr  \" 5       (       a  S SKJrJr  S SKJrJ	r	  Sr
S rS rS	 rS
 rS rg)    N   )is_sklearn_availablerequires_backends)pearsonr	spearmanr)f1_scorematthews_corrcoefu  This metric will be removed from the library soon, metrics should be handled with the 🤗 Evaluate library. You can have a look at this example script for pointers: https://github.com/huggingface/transformers/blob/main/examples/pytorch/text-classification/run_glue.pyc                     [         R                  " [        [        5        [	        [
        S5        X:H  R                  5       $ )Nsklearn)warningswarnDEPRECATION_WARNINGFutureWarningr   simple_accuracymean)predslabelss     \/home/james-whalen/.local/lib/python3.13/site-packages/transformers/data/metrics/__init__.pyr   r      s,    MM%}5oy1O!!##    c                     [         R                  " [        [        5        [	        [
        S5        [        X5      n[        XS9nUUX#-   S-  S.$ )Nr   )y_truey_pred   )accf1
acc_and_f1)r   r   r   r   r   r   r   r   )r   r   r   r   s       r   r   r   $   sH    MM%}5j),
%
(C		.Bx1n r   c                     [         R                  " [        [        5        [	        [
        S5        [        X5      S   n[        X5      S   nUUX#-   S-  S.$ )Nr   r   r   )pearsonr   corr)r   r   r   r   r   pearson_and_spearmanr   r   )r   r   pearson_corrspearman_corrs       r   r    r    0   sU    MM%}5*I6E*1-Le,Q/M"-2 r   c                 z   [         R                  " [        [        5        [	        [
        S5        [        U5      [        U5      :X  d   S[        U5       S[        U5       35       eU S:X  a  S[        X!5      0$ U S:X  a  S[        X5      0$ U S:X  a  [        X5      $ U S	:X  a  [        X5      $ U S
:X  a  [        X5      $ U S:X  a  S[        X5      0$ U S:X  a  S[        X5      0$ U S:X  a  S[        X5      0$ U S:X  a  S[        X5      0$ U S:X  a  S[        X5      0$ U S:X  a  S[        X5      0$ [        U 5      e)Nr   /Predictions and labels have mismatched lengths  and colamcczsst-2r   mrpczsts-bqqpmnlizmnli/acczmnli-mmzmnli-mm/accqnlirtewnlihans)r   r   r   r   r   glue_compute_metricslenr	   r   r   r    KeyError	task_namer   r   s      r   r/   r/   <   s[   MM%}5*I6u:V$v(WX[\aXbWcchilmsithu&vv$F(788	g	u566	f	%((	g	#E22	e	%((	f	OE:;;	i	u=>>	f	u566	e	u566	f	u566	f	u566y!!r   c                    [         R                  " [        [        5        [	        [
        S5        [        U5      [        U5      :w  a#  [        S[        U5       S[        U5       35      eU S:X  a  S[        X5      0$ [        U 5      e)Nr   r$   r%   xnlir   )
r   r   r   r   r   xnli_compute_metricsr0   
ValueErrorr   r1   r2   s      r   r6   r6   Z   su    MM%}5*I6
5zS[ J3u:,V[\_`f\g[hijjFu566y!!r   )r   utilsr   r   scipy.statsr   r   sklearn.metricsr   r	   r   r   r   r    r/   r6    r   r   <module>r<      sB     < /;m $		"<"r   