hoops_ai.dataset.CADDataset

class hoops_ai.dataset.CADDataset(parent_dataset, indices)

Bases: object

A framework-agnostic dataset object that contains training, validation or testing data. Can be converted to framework-specific formats as needed.

get_item(i)

Framework-agnostic item access

get_raw_data(i)

Get raw file paths for an item without loading

remove_indices(local_indices_to_remove)

Remove items by local subset index. This also removes them from the parent dataset, so the parent’s data_files/label_datas arrays and indexing will be updated. Then we adjust self.indices accordingly.

to_torch(collate_fn=None)

Convert to PyTorch Dataset using lazy import

Parameters:

collate_fn – Optional collate function for batching

property data_files

Return the list of .bin file paths for this subset only.

property label_datas

Return the list of label arrays for this subset only.