kosmos.ml.datasets.organamnist_dataset

Classes

class OrganAMNISTDataset(*, min_max_scaler: bool = True)

Bases: kosmos.ml.datasets.dataset.SLDataset

OrganAMNIST dataset for organ classification from medical images.

Notes

  • Number of instances: 58,830 (34,561 train + 6,491 val + 17,778 test)

  • Number of features: 784 numeric (28x28 pixel images, flattened)

  • Classes: 11 (different organ types)

References

Initialize the dataset.

Parameters:

min_max_scaler (bool) – Whether to apply min-max scaling to the features.


Properties

property class_names

Return human-readable class labels for organ types.

property input_dimension

Number of feature columns.

property output_dim

Number of distinct classes.