
    RЦi                     &    S r  " S S5      rSS jrg)zLEval metrics and related

Hacked together by / Copyright 2020 Ross Wightman
c                   .    \ rS rSrSrS rS rSS jrSrg)	AverageMeter   z1Computes and stores the average and current valuec                 $    U R                  5         g N)resetselfs    Q/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/timm/utils/metrics.py__init__AverageMeter.__init__	   s    

    c                 <    SU l         SU l        SU l        SU l        g )N    )valavgsumcountr   s    r
   r   AverageMeter.reset   s    
r   c                     Xl         U =R                  X-  -  sl        U =R                  U-  sl        U R                  U R                  -  U l        g r   )r   r   r   r   )r	   r   ns      r
   updateAverageMeter.update   s8    CG

a
88djj(r   )r   r   r   r   N   )	__name__
__module____qualname____firstlineno____doc__r   r   r   __static_attributes__ r   r
   r   r      s    ;)r   r   c           	         [        [        U5      U R                  5       S   5      nUR                  S5      nU R                  USSS5      u  pVUR	                  5       nUR                  UR                  SS5      R                  U5      5      nU Vs/ s HB  oS[        X5       R                  S5      R                  5       R                  S5      S-  U-  PMD     sn$ s  snf )zNComputes the accuracy over the k top predictions for the specified values of kr   r   TNg      Y@)
minmaxsizetopkteqreshape	expand_asfloatr   )	outputtargetr'   maxk
batch_size_predcorrectks	            r
   accuracyr5      s    s4y&++-*+DQJkk$4.GA668DggfnnQ+55d;<G[_`[_VWMS\"**2.446::1=DzQ[_```s   A	CN)r   )r   r   r5   r!   r   r
   <module>r6      s   ) )$ar   