wvcg0mt7pkASSI 3 ML 16
wvcg0mt7pkASSI 3 ML 16
SRN No 202101542
Roll No 16
Division B
Assignment 3
Create Multi variable Regression model of your choice using suitable dataset.
Source Code:
import numpy as np
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error, r2_score
import matplotlib.pyplot as plt
import seaborn as sns
# Convert to DataFrame
data = pd.DataFrame(housing.data, columns=housing.feature_names)
data['MedHouseVal'] = housing.target
# Correlation heatmap
plt.figure(figsize=(10, 8))
sns.heatmap(data.corr(), annot=True, cmap='coolwarm')
plt.title("Correlation Heatmap")
plt.show()
# Prepare the data for training
X = data.drop('MedHouseVal', axis=1)
y = data['MedHouseVal']
# Residual plot
sns.residplot(y_test, y_pred)
plt.title("Residual Plot")
plt.show()
OUTPUT :-