DfBatchGenerator

class lightautoml.reader.tabular_batch_generator.DfBatchGenerator(data, n_jobs=1, batch_size=None)[source]

Bases: BatchGenerator

Batch generator from DataFrame.

Parameters:

  • data (DataFrame) – Data used for generator.

  • n_jobs (int) – Number of processes to handle.

  • batch_size (Optional[int]) – Batch size. Default is None, split by n_jobs.