メインコンテンツまでスキップ

make_batch

make_batch(data: Union[Iterable, Generator], batch_size: int) -> Generator[List, None, None]

  • 説明:この関数はデータをバッチデータに変換するために使用されます。

  • パラメータ

    • data (Union[Iterable, Generator]):データジェネレーター。
    • batch_size (int):バッチデータのサイズ。
  • 戻り値

    • Generator[List, None, None]:バッチデータのジェネレーター。
  • import docsaidkit as D

    data = range(10)
    batch_size = 3
    batched_data = D.make_batch(data, batch_size)
    for batch in batched_data:
    print(batch)
    # >>> [0, 1, 2]
    # >>> [3, 4, 5]
    # >>> [6, 7, 8]
    # >>> [9]