cellmap_data.subdataset

Contents

cellmap_data.subdataset#

Classes

CellMapSubset(dataset, indices)

class cellmap_data.subdataset.CellMapSubset(dataset: CellMapDataset | CellMapMultiDataset, indices: Sequence[int])[source]#

Parameters:

dataset (CellMapDataset | CellMapMultiDataset) – CellMapDataset | CellMapMultiDataset The dataset to be subsetted.
indices (Sequence[int]) – Sequence[int] The indices of the dataset to be used as the subset.

property classes: Sequence[str]#: The classes in the dataset.

property class_counts: dict[str, float]#: The number of samples in each class in the dataset normalized by resolution.

property class_weights: dict[str, float]#: The class weights for the dataset based on the number of samples in each class.

property validation_indices: Sequence[int]#: The indices of the validation set.

to(device) → CellMapSubset[source]#

Move the dataset to the specified device.

Return type:: CellMapSubset

set_raw_value_transforms(transforms: Callable) → None[source]#

Sets the raw value transforms for the subset dataset.

Parameters:: transforms (Callable)
Return type:: None

set_target_value_transforms(transforms: Callable) → None[source]#

Sets the target value transforms for the subset dataset.

Parameters:: transforms (Callable)
Return type:: None

dataset: Dataset[_T_co]#

indices: Sequence[int]#