Scribd
Upload
18 views2 pages

Experiment 5

The document outlines a machine learning workflow using Python to predict purchases based on age and estimated salary from a dataset. It includes data loading, preprocessing, model training with logistic regression, and evaluation using a confusion matrix and various metrics such as accuracy, precision, and recall. The results indicate an accuracy of 86% with a precision of 90% and a recall of 68%.

Uploaded by

dirilan857
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd
18 views2 pages

Experiment 5

The document outlines a machine learning workflow using Python to predict purchases based on age and estimated salary from a dataset. It includes data loading, preprocessing, model training with logistic regression, and evaluation using a confusion matrix and various metrics such as accuracy, precision, and recall. The results indicate an accuracy of 86% with a precision of 90% and a recall of 68%.

Uploaded by

dirilan857
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd
You are on page 1/ 2

In [1]: import pandas as pd

import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns

from sklearn.model_selection import train_test_split


from sklearn.preprocessing import StandardScaler
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import confusion_matrix, accuracy_score, precision_score, recall_score

In [9]: df = pd.read_csv(r'C:\Users\VEDANT\Desktop\Data sci\Social_Network_Ads\Social_Network_Ads.csv


print(df.head())

Age EstimatedSalary Purchased


0 19 19000 0
1 35 20000 0
2 26 43000 0
3 27 57000 0
4 19 76000 0

In [11]: X = df[['Age', 'EstimatedSalary']]


y = df['Purchased']

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

# Feature scaling (important for LR)


scaler = StandardScaler()
X_train = scaler.fit_transform(X_train)
X_test = scaler.transform(X_test)

In [12]: model = LogisticRegression()


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

In [13]: # Confusion matrix


cm = confusion_matrix(y_test, y_pred)
tn, fp, fn, tp = cm.ravel()

# Metrics
accuracy = accuracy_score(y_test, y_pred)
error_rate = 1 - accuracy
precision = precision_score(y_test, y_pred)
recall = recall_score(y_test, y_pred)

print("Confusion Matrix:")
print(cm)
print(f"True Positives: {tp}")
print(f"False Positives: {fp}")
print(f"True Negatives: {tn}")
print(f"False Negatives: {fn}")
print("\n--- Metrics ---")
print(f"Accuracy: {accuracy:.2f}")
print(f"Error Rate: {error_rate:.2f}")
print(f"Precision: {precision:.2f}")
print(f"Recall: {recall:.2f}")
Confusion Matrix:
[[50 2]
[ 9 19]]
True Positives: 19
False Positives: 2
True Negatives: 50
False Negatives: 9

--- Metrics ---


Accuracy: 0.86
Error Rate: 0.14
Precision: 0.90
Recall: 0.68

In [14]: sns.heatmap(cm, annot=True, fmt='d', cmap='Blues')


plt.title("Confusion Matrix")
plt.xlabel("Predicted")
plt.ylabel("Actual")
plt.show()

You might also like