hoops_ai.dataset.TorchCADDataset

class hoops_ai.dataset.TorchCADDataset(*args, **kwargs)

Bases: Dataset

PyTorch Dataset wrapper for CADDataset

get_dataloader(batch_size=64, shuffle=False, num_workers=0)

Create a DataLoader for this dataset

Parameters:

batch_size (int)

remove_indices(local_indices_to_remove)

Remove items by delegating to the underlying CADDataset

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.