o 2h(@s^ddlZddlZddlZddlZddlmZddlmZddlm Z edGdddZ dS)N)backend) keras_export)io_utilszkeras.utils.Progbarc@sHeZdZdZ     dddZdd d Zdd d Zd dZddZdS)Progbara_Displays a progress bar. Args: target: Total number of steps expected, None if unknown. width: Progress bar width on screen. verbose: Verbosity mode, 0 (silent), 1 (verbose), 2 (semi-verbose) stateful_metrics: Iterable of string names of metrics that should *not* be averaged over time. Metrics in this list will be displayed as-is. All others will be averaged by the progbar before display. interval: Minimum visual progress update interval (in seconds). unit_name: Display name for step counts (usually "step" or "sample"). 皙?NstepcCs||_||_||_||_||_|rt||_nt|_ttj dr&tj p4dtj vp4dtj vp4dt j v|_d|_i|_g|_t|_d|_|j|_d|_d|_dS)Nisatty ipykernelposixPYCHARM_HOSTEDr)targetwidthverboseinterval unit_namesetstateful_metricshasattrsysstdoutr modulesosenviron_dynamic_display _seen_so_far_values _values_ordertime_start _last_update_time_at_epoch_start_time_after_first_step_prev_total_width)selfrrrrrrr&R/var/www/html/chatgem/venv/lib/python3.10/site-packages/keras/src/utils/progbar.py__init__s.    zProgbar.__init__c Cs|dur|jdur d}n||jk}|pg}|D]U\}}||jvr&|j|||jvre|r5|dg|j|<qt||jd}||jvrL|||g|j|<q|j|d||7<|j|d|7<q|dg|j|<q||_d}d}t} ||| } |j dkr| |j |j kr|sdS|j r|d|j 7}|d7}n|d7}|jdurtt|jd} d t| d ||jf} d | d } |d 7}t||j} t|j| }|dkr| dd|d7} |d7}| dd|j|d7} |d7}nd|} || 7}|jdurC|sC| |j|}|dkr(d|d|dd|df}n|dkr8d|d|df}nd|}d|d}n d| |jdd}|d 7}||| |j7}|jD]U}|d|d7}t|j|trttj|j|dtd|j|d}t|}t|dkr|d|d 7}q]|d|d!7}q]|d|j|7}q]||7}t| t||}|j |kr|d|j |7}|r|d7}tj |dd"||_ d}n|j d#krj|rjtt|jd} d t| d ||jf}|d| |jdd$}|d%|| |j7}|jD]:}|d|d7}ttj|j|dtd|j|d}|dkrO|d|d 7}q|d|d!7}q|d7}||7}tj |dd"d}| |_ dS)&aUpdates the progress bar. Args: current: Index of current step. values: List of tuples: `(name, value_for_last_step)`. If `name` is in `stateful_metrics`, `value_for_last_step` will be displayed as-is. Else, an average of the metric over time will be displayed. finalize: Whether this is the last update for the progress bar. If `None`, defaults to `current >= self.target`. NFrr  %zd/%dzz zu━z zz %7d/Unknowniz %d:%02d:%02d<z%d:%02dz%dsz .0fzsz - :MbP? z.4fz.4e) line_breaksz -)!rrappendrrmaxrr_estimate_step_durationrr!rrr$intmathlog10strfloatrr _format_timer isinstancelistrconvert_to_numpynumpymeanabslenr print_msg)r%currentvaluesfinalizekv value_basemessagespecial_char_lennow time_per_unit numdigitsbarprog prog_widtheta eta_formatinfoavg total_widthcountr&r&r'update>s                        zProgbar.updatecCs||j||dSN)r]r)r%nrJr&r&r'addsz Progbar.addcCsrd}|dks |dkr|d|dd|7}|S|dkr*|d|ddd |7}|S|d|d dd |7}|S) aformat a given duration to display to the user. Given the duration, this function formats it in either milliseconds or seconds and displays the unit (i.e. ms/step or s/epoch). Args: time_per_unit: the duration to display unit_name: the name of the unit to display Returns: A string with the correctly formatted duration and units r)rrr4r1zs/r3g@@zms/g.Azus/r&)r%rRr formattedr&r&r'r@s zProgbar._format_timecCsN|r%|jdur|dkr||j|d}n||j|}|dkr#||_|SdS)aEstimate the duration of a single step. Given the step number `current` and the corresponding time `now` this function returns an estimate for how long a single step takes. If this is called before one step has been completed (i.e. `current == 0`) then zero is given as an estimate. The duration estimate ignores the duration of the (assumed to be non-representative) first step for estimates when more steps are available (i.e. `current>1`). Args: current: Index of current step. now: The current time. Returns: Estimate of the duration of a single step. Nrr)r#r )r%rIrQrRr&r&r'r:szProgbar._estimate_step_duration)rrrNr )NNr^) __name__ __module__ __qualname____doc__r(r]r`r@r:r&r&r&r'r s $  r) r<rrr keras.srcrkeras.src.api_exportrkeras.src.utilsrrr&r&r&r's