tinker_cookbook.supervised.SupervisedDatasetFromHFDataset

class tinker_cookbook.supervised.SupervisedDatasetFromHFDataset( SupervisedDataset)

A supervised dataset backed by a HuggingFace dataset.

get_batch( index)

Return a batch of Datum objects at the given index.

Parameters:

• index ( int) – Zero-based batch index.

Returns: list[tinker. Datum – Training datums for this batch.

set_epoch( seed)

Shuffle the dataset for a new epoch.

Parameters:

• seed ( int) – Random seed for shuffling. Default 0.