tinker_cookbook.supervised.SupervisedDatasetBuilder

class tinker_cookbook.supervised.SupervisedDatasetBuilder(**)

A chz config class that knows how to construct a supervised dataset.

__call__()

Build and return (train_dataset, eval_dataset).

Returns: tuple[SupervisedDataset, SupervisedDataset | None]: The training dataset and an optional evaluation dataset.