lczerolens.game.dataset#

Dataset class for lczero models.

Classes#

GameDataset: A class for representing a dataset of games.
BoardDataset: A class for representing a dataset of boards.
IterableBoardDataset: A class for representing an iterable dataset of boards.

class lczerolens.GameDataset(file_name: str | None = None, games: List[Game] | None = None)#

A class for representing a dataset of games.

games#

The list of games.

class lczerolens.BoardDataset(file_name: str | None = None, boards: List[Board] | None = None, game_ids: List[str] | None = None)#

A class for representing a dataset of boards.

boards#

The list of boards.

save(file_name: str, n_history: int = 0, indices=None)#: Save the dataset to a file.

Note

As the board needs to be unpiled use the preprocess_game method.