Page 292 - Touhpad Ai

PROJECT WORK

Project 1

This is a Placement Prediction Model that uses Linear Regression to predict a student’s placement package (salary)
based on their CGPA. Download the dataset from https://www.kaggle.com/datasets/parvmodi/cgpa-vs-package-in-lpa
or scan the QR code.
# Step 1: Import necessary libraries
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
from sklearn.linear_model import LinearRegression

# Step 2: Load and clean the dataset
df = pd.read_csv("placement.csv")
df = df.dropna()

# Step 3: Visualize CGPA vs Package
plt.figure(figsize=(6, 4))
sns.scatterplot(x='cgpa', y='package', data=df, label='Actual Data')

# Step 4: Linear Regression Model
X = df[['cgpa']]
y = df['package']

model = LinearRegression()
model.fit(X, y)

# Plot regression line
predicted_packages = model.predict(X)
plt.plot(df['cgpa'], predicted_packages, color='red', label='Regression Line')
plt.title('CGPA vs Placement Package')
plt.xlabel('CGPA')
plt.ylabel('Package (in LPA)')
plt.legend()
plt.grid(True)

290 Touchpad Artificial Intelligence - XI