Notice: This page requires JavaScript to function properly.
Please enable JavaScript in your browser settings or update your browser.
Lära Class Balance | Recognizing Handwritten Digits
Recognizing Handwritten Digits

book
Class Balance

Ensuring class balance in machine learning datasets is crucial for avoiding model bias towards any specific class. When a dataset is balanced, it signifies an equitable representation of all classes, which is vital. An imbalanced dataset can result in suboptimal performance of the model, especially in predicting the minority class.

Consider a dataset comprising customer transactions where 90% are legitimate and only 10% fraudulent. Training a model on such skewed data might incline it to predict most transactions as legitimate. This inclination occurs because the model is tailored to reduce overall error, and identifying most transactions as legitimate boosts accuracy, albeit superficially.

Hence, maintaining class balance is imperative for training models on a diverse sample from each class, enhancing their ability to make precise predictions across the board.

Uppgift

Swipe to start coding

  1. Extract the target labels (digit values) from the MNIST dataset and assign them to variable Y.

  2. Visualize the frequency distribution of digit labels from the MNIST dataset by creating a count plot.

Lösning

import pandas as pd

# Extract the target labels (digit values) from the MNIST dataset
Y = mnist['target']

# Create a figure and axis with specific size for plotting the digit label distribution
f, ax = plt.subplots(figsize=(8, 3))
plt.title('Distribution of Digit Labels')

# Plot a count plot of digit frequencies
sns.countplot(x=Y, order=pd.Series(Y).value_counts().index)

# Annotate bars with the count of each digit label
ax.bar_label(container=ax.containers[0])

# Set the y-axis limit and label the x-axis
ax.set(ylim=[0, 9000], xlabel='Digit')
plt.show()

Mark tasks as Completed
Var allt tydligt?

Hur kan vi förbättra det?

Tack för dina kommentarer!

Avsnitt 1. Kapitel 4
AVAILABLE TO ULTIMATE ONLY
some-alt