o
    7ti                     @   s   d dl Zdd ZdS )    Nc                 C   s|   ddl m} t|  \}}t|}t|}||dk|dkd}||dk|dkd}||dk|dkd}t|||g}|S )Nr   )f1_score)y_truey_pred      )sklearn.metricsr   zipnparraymean)itemsr   predsgoldsf11f12f13avg_f1 r   Y/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/super_glue/cb/aggregate.pycb_multi_fi   s   

r   )numpyr	   r   r   r   r   r   <module>   s    