&VfS8dZddlZddlmZGddZdS)a- Separation of concerns: DataAdapter: - x, y - sample_weight - class_weight - shuffle - batch_size - steps, as it relates to batch_size for array data EpochIterator: - whether to yield numpy or tf data - steps - most argument validation Trainer: - steps_per_execution - validation_split - validation_data - callbacks - validation_freq - epochs - initial_epoch - any backend-specific concern such as distribution PyDataset: - num_workers - use_multiprocessing - max_queue_size EpochIterator steps: 1. Look at data type and select correct DataHandler 2. Instantiate DataHandler with correct arguments 3. Raise or warn on unused arguments 4. in __iter__, iterate, either for a fixed number of steps or until there is no data N) data_adapterscFeZdZ ddZdZdZedZdS) EpochIteratorNFc ||_||_|rd|_d|_t j||||||||_|jj|_dS)NF)xy sample_weight batch_sizesteps_per_epochshuffle class_weight) r steps_per_execution_current_iterator_insufficient_datarget_data_adapter data_adapter num_batches _num_batches) selfrr r r r r rrs ^/var/www/html/software/conda/lib/python3.11/site-packages/keras/src/trainers/epoch_iterator.py__init__zEpochIterator.__init__0sq /#6  ,%)D "&+D #):'!+%   !-9c4|jSN)rget_numpy_iteratorrs r _get_iteratorzEpochIterator._get_iteratorKs 33555rc#Kg}|jr|j-t||_d|_t |jD]}|jrn t |j}||t||j kr|t|z dz|fVg}i#tf$r'tj ddd|_d|_YwxYw|r|t|z dz|fVnt|D]M\}}||t||j kr|t|z dz|fVg}N|r|t|z dz|fV|js |dz|_|jdS)NFrzYour input ran out of data; interrupting epoch. Make sure that your dataset or generator can generate at least `steps_per_epoch * epochs` batches. You may need to use the `.repeat()` function when building your dataset.) stacklevelT)r riterrrrangenextappendlenr StopIterationwarningswarn enumeraterr on_epoch_end)rbufferstepdatas renumerate_epochzEpochIterator.enumerate_epochNs%  ' -%-)-d.@.@.B.B)C)C&*/'d233 3 3*E3 677DMM$'''6{{d&>>>"S[[014f<<<>/C0/C0c,|jr|jS|jSr)r rrs rrzEpochIterator.num_batcheszs!   (' '  r)NNNNFNr)__name__ __module__ __qualname__rrr/propertyrrrrr/sy ::::6666*)*)*)X!!X!!!rr)__doc__r(keras.src.trainersrrr5rrr8si''R,,,,,,Q!Q!Q!Q!Q!Q!Q!Q!Q!Q!r