Why classification predicts categories in ML Python

Introduction

Classification helps us sort things into groups or categories. It predicts which group new data belongs to, making decisions easier.

Sorting emails into 'spam' or 'not spam'.

Recognizing if a photo shows a cat, dog, or bird.

Deciding if a bank transaction is 'fraud' or 'safe'.

Classifying customer reviews as 'positive' or 'negative'.

Identifying handwritten digits from 0 to 9.

Syntax

ML Python

model.fit(X_train, y_train)
predictions = model.predict(X_test)

fit() trains the model using input data and known categories.

predict() uses the trained model to guess categories for new data.

Examples

Train a decision tree to classify data into categories.

ML Python

from sklearn.tree import DecisionTreeClassifier
model = DecisionTreeClassifier()
model.fit(X_train, y_train)
predictions = model.predict(X_test)

Use logistic regression to predict categories like 'yes' or 'no'.

ML Python

from sklearn.linear_model import LogisticRegression
model = LogisticRegression()
model.fit(X_train, y_train)
predictions = model.predict(X_test)

Sample Program

This program trains a decision tree to classify iris flowers into species. It shows the predicted categories and how accurate the model is.

ML Python

from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier
from sklearn.metrics import accuracy_score

# Load example data
iris = load_iris()
X, y = iris.data, iris.target

# Split data into training and testing sets
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)

# Create and train the model
model = DecisionTreeClassifier(random_state=42)
model.fit(X_train, y_train)

# Predict categories for test data
predictions = model.predict(X_test)

# Calculate accuracy
accuracy = accuracy_score(y_test, predictions)

print(f"Predictions: {predictions}")
print(f"Accuracy: {accuracy:.2f}")

Important Notes

Classification models learn from examples with known categories.

Accuracy shows how often the model guesses correctly.

Categories must be clearly defined before training.

Summary

Classification predicts which group new data belongs to.

It is useful for sorting and decision-making tasks.

Models learn by training on labeled examples.