Name the Classes

10 pts · 10 coins

Machine Learning in CodeEasy

A built-in dataset stores its labels as numbers: 0, 1, 2. The real names live in a separate key.

Task: write class_counts(name). name is one of "iris", "wine" or "breast_cancer".

Load the dataset and count how many rows have each label.
Return a dict that maps each label's name (a string from target_names) to its count.
Label number i has the name at position i of target_names.

name = "iris"

returns {'setosa': 50, 'versicolor': 50, 'virginica': 50}

name = "breast_cancer"

returns {'malignant': 212, 'benign': 357}

solution.py

1from sklearn.datasets import load_iris, load_wine, load_breast_cancer
2
3
4def class_counts(name):
5    pass
6