Scribd
Upload
7 views1 page

1 - Linear Regression With MSE Function PLot

The document contains a Python script that demonstrates a simple linear regression using Matplotlib and Scikit-learn. It generates random data, fits a linear regression model, calculates the Mean Squared Error, and visualizes the data points along with the regression line and errors. Additionally, it suggests exploring a similar graph on Desmos for further analysis of minima through differentiation.

Uploaded by

sambha7896
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
7 views1 page

1 - Linear Regression With MSE Function PLot

The document contains a Python script that demonstrates a simple linear regression using Matplotlib and Scikit-learn. It generates random data, fits a linear regression model, calculates the Mean Squared Error, and visualizes the data points along with the regression line and errors. Additionally, it suggests exploring a similar graph on Desmos for further analysis of minima through differentiation.

Uploaded by

sambha7896
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/ 1

1/4/25, 11:34 AM Function_OLS_plot.

ipynb - Colab

import matplotlib.pyplot as plt


import numpy as np
from sklearn.linear_model import LinearRegression

# Creating a simple linear regression example


np.random.seed(0)
x = 2 * np.random.rand(100, 1)
y = 4 + 3 * x + np.random.randn(100, 1)

# Fitting the linear regression model


lin_reg = LinearRegression()
lin_reg.fit(x, y)
y_pred = lin_reg.predict(x)

# Calculating the Mean Squared Error -> This is what needs to be minimized for the line of best fit
#What can be the possible y=f(x) relationship if x denotes the error y is the outcome
mse = np.mean((y - y_pred)**2)

# Plotting the data points


plt.scatter(x, y, color='blue', label='Data points')

# Plotting the linear regression line


plt.plot(x, y_pred, color='red', label='Linear regression line')

# Plotting the errors (distances from the points to the line)


for i in range(len(x)):
plt.plot([x[i], x[i]], [y[i], y_pred[i]], color='green', linestyle='--')

# Adding labels and title


plt.xlabel('x')
plt.ylabel('y')
plt.title(f'Linear Regression with MSE: {mse:.2f}')
plt.legend()

# Show the plot


plt.show()

#Now let's plot a similar graph on the https://www.desmos.com/calculator


# and see if it is possible to find a minima of this -> We will explore this more in Differentiation

https://colab.research.google.com/drive/16uRXrSm8m0ARl7A9kJfBpeuVdLzYGUie#printMode=true 1/2

You might also like