keras.cpython-312.pyc•6.91 kB
�
lMg � �� � d dl m Z d dlmZ ddlmZ d dlZddgiZ d gZ G d
� d ej j � Zy# eef$ r Z d dl
mZ n # e$ r e �w xY wY dZ [ �GdZ [ ww xY w) � )�copy)�partial� )�tqdmN)�kerasz github.com/� casperdcl� TqdmCallbackc � � e Zd ZdZedd� fd�� Zddddefd�Zd� Zd� Z d � Z
d
� Z ed � � Z ed � � Z
ed
� � Zy)r z,Keras callback for epoch and batch progress.Nc � � y)Nr � ��logss �VC:\Users\noahv\Documents\GitHub\clickup-operator\.venv\Lib\site-packages\tqdm/keras.py�<lambda>zTqdmCallback.<lambda> s � �� c � � ��� d� ��fd� }|S )Nc �� �� �|� }|r@�r+t |� }�D �cg c] }|j |d� �� c} �j |d�� �j |� y c c}w )Nr F)�refresh)r �pop� set_postfix�update)�_r �n�i�bar�deltar s ���r �callbackz+TqdmCallback.bar2callback.<locals>.callback sW �� ��d� �A��� ��:�D�-0�1�S��T�X�X�a��^�S�1�����e��4� �J�J�q�M�� 2s �A )Nr )r r r r s ``` r � bar2callbackzTqdmCallback.bar2callback s �� � � �r r c �R � |r t |fi |��}|| _ ||d�� | _ | j | j � | _ |r|r||z dz
|z x| _ }n dx| _ }|| _ |dk( r7 ||dd�� | _ | j | j ddgd � �
� | _ yy) a^
Parameters
----------
epochs : int, optional
data_size : int, optional
Number of training pairs.
batch_size : int, optional
Number of training pairs per batch.
verbose : int
0: epoch, 1: batch (transient), 2: batch. [default: 1].
Will be set to `0` unless both `data_size` and `batch_size`
are given.
tqdm_class : optional
`tqdm` class to use for bars [default: `tqdm.auto.tqdm`].
tqdm_kwargs : optional
Any other arguments used for all bars.
�epoch)�total�unitr N�batchF)r! r" �leave�sizec �& � | j dd� S �Nr% r ��getr
s r r z'TqdmCallback.__init__.<locals>.<lambda>A s � �4�8�8�F�A�#6r �r r ) r �
tqdm_class� epoch_barr � on_epoch_end�batches�verbose� batch_bar� on_batch_end)�self�epochs� data_size�
batch_sizer/ r+ � tqdm_kwargsr. s r �__init__zTqdmCallback.__init__ s� � �& � ��;�{�;�J�$���#�&�w�?��� � -� -�d�n�n� =��� ��&/�*�&<�q�&@�Z�%O� O�D�L�7�%)� )�D�L�7� �� � �a�<� '�g�G�5� Q�D�N� $� 1� 1����W�f�$5�6� !2� !8�D� � r c �� � | j j } |d |dd � � }|�7|| j j k7 r | j j |�� y y y )Nr3 �nb_epoch�r! )�paramsr) r, r! �reset)r2 r �__r; �
auto_totals r �on_train_beginzTqdmCallback.on_train_beginC sW � ���� � ���H�f�Z��&>�?�
� � !�j�D�N�N�4H�4H�&H� �N�N� � �z� � 2� 'I� !r c
� � | j j |k r!| j }|x|_ x|_ |_ | j �r| j
j } |d |d |dd � � � xs | j }| j dk( rut | d� r| j j � | j |ddd |d d� xs dz �
� | _ | j | j dd gd � �
� | _
y | j dk( r<d |d d� xs dz | j _ | j j |�� y t! d� �y )N�samples� nb_sample�steps� r0 r# Tr r5 )r! r" r$ �
unit_scaler% c �& � | j dd� S r' r( r
s r r z-TqdmCallback.on_epoch_begin.<locals>.<lambda>Y s � �t�x�x���':r r* r: zUnknown verbosity)r, r � last_print_n�initialr/ r; r) r. �hasattrr0 �closer+ r r1 rE r< �KeyError)r2 r r r= �ebarr; r! s r �on_epoch_beginzTqdmCallback.on_epoch_beginI sI � � �>�>� � �e� #��>�>�D�8=� =�D�F� =�T�&�� � �<�<��[�[�_�_�F��9�f��V�G�T� 2�'4� 5� E�8<� � �
��|�|�q� ��4��-��N�N�(�(�*�!%� � � �g�T� �F�<��$;�$@�q� A� "1� "C��� %)�$5�$5��N�N��&�(9�:� %6� %<��!� � � ��"�,-�� �a�1H�1M�A�,N����)����$�$�5�$�1� � 2�3�3�# r c � � t | d� r| j j � | j j � y )Nr0 )rI r0 rJ r, )r2 r r= s r � on_train_endzTqdmCallback.on_train_end` s, � � �4�� %� �N�N� � � "� ��� � � r c � � t | j dd� }|�yddlm} ||� t | dd� }|� ||j � yy)z*Displays in the current cell in Notebooks.� containerNr )�displayr0 )�getattrr, �notebookrR rQ )r2 rQ rR r0 s r rR zTqdmCallback.displaye sO � ��D�N�N�K��>� � � � �%�� ���D�+�t�4� � � � �I�'�'� (� !r c � � y�NTr r r r � _implements_train_batch_hooksz*TqdmCallback._implements_train_batch_hooksp � � �r c � � yrV r r r r � _implements_test_batch_hooksz)TqdmCallback._implements_test_batch_hookst rX r c � � yrV r r r r � _implements_predict_batch_hooksz,TqdmCallback._implements_predict_batch_hooksx rX r )�__name__�
__module__� __qualname__�__doc__� staticmethodr � tqdm_autor7 r? rM rO rR rW rZ r\ r r r r r s� � �6�� "�>�
� �
� #�d�t�Q� %�!8�F3� 4�. �
)� �� �� �� �� �� �r )r � functoolsr �autor rb r � ImportError�AttributeError�e�
tensorflow�
__author__�__all__� callbacks�Callbackr r r r �<module>rm s| �� � � #��� �k�]�
+�
� �
��i�5�?�?�+�+� i�� �^�$� ��$�� � ����� %���s) �A � A$�A�A �A�A � A$