sup3r.preprocessing.batch_handlers.factory.BatchHandlerFactory

sup3r.preprocessing.batch_handlers.factory.BatchHandlerFactory#

BatchHandlerFactory(MainQueueClass, SamplerClass, ValQueueClass=None, name='BatchHandler')[source]#

BatchHandler factory. Can build handlers from different queue classes and sampler classes. For example, to build a standard BatchHandler use SingleBatchQueue and Sampler. To build a DualBatchHandler use DualBatchQueue and DualSampler. To build a BatchHandlerDC use a BatchQueueDC, ValBatchQueueDC and SamplerDC

Note

(1) BatchHandlers include a queue for training samples and a queue for validation samples. (2) There is no need to generate “Spatial” batch handlers. Using Sampler objects with a single time step in the sample shape will produce batches without a time dimension.