kosmos.ml.datasets.wdbc_dataset

Classes

class WDBCDataset(*, min_max_scaler: bool = True)

Bases: kosmos.ml.datasets.dataset.SLDataset

WDBC (wisconsin breast cancer) dataset for binary classification.

Notes

  • Number of instances: 569

  • Number of features: 30 numeric

  • Classes: 2 (slighty imbalanced; 357 benign, 212 malign)

References

Initialize the dataset.

Parameters:

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


Properties

property class_names

B->0 (Benign), M->1 (Malignant).

Type:

Return human-readable class labels and map

property input_dimension

Number of feature columns.

property output_dim

Number of distinct classes.