In this lab, you'll get some hand-on practice developing a simple linear regression model. You'll also use your model to make a prediction about new data!
You will be able to:
- Perform a linear regression using self-constructed functions
- Interpret the parameters of a simple linear regression model in relation to what they signify for specific data
The best-fit line's slope
With
You'll use the latter formula in this lab. First, break down the formula into its parts. To do this, you'll import the required libraries and define some data points to work with. Next, you'll use some pre-created toy data in NumPy arrays. Let's do this for you to give you a head start.
# import necessary libraries
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import style
style.use('ggplot')
%matplotlib inline
# Initialize arrays X and Y with given values
# X = Independent Variable
X = np.array([1,2,3,4,5,6,8,8,9,10], dtype=np.float64)
# Y = Dependent Variable
Y = np.array([7,7,8,9,9,10,10,11,11,12], dtype=np.float64)
# Scatter plot
# Your observations about the relationship between X and Y
#
Write a function calc_slope()
that takes in X and Y and calculates the slope using the formula shown above.
# Write the function to calculate slope as:
# (mean(x) * mean(y) – mean(x*y)) / ( mean (x)^2 – mean( x^2))
def calc_slope(xs,ys):
pass
calc_slope(X,Y)
# 0.5393518518518512
Great, so we have our slope. Next we calculate the intercept.
As a reminder, the calculation for the best-fit line's y-intercept is:
Write a function best_fit()
that takes in X and Y, calculates the slope and intercept using the formula. The function should return slope and intercept values.
# use the slope function with intercept formula to return calculate slope and intercept from data points
def best_fit(xs,ys):
pass
# Uncomment below to test your function
#m, c = best_fit(X,Y)
#m, c
# (0.5393518518518512, 6.379629629629633)
We now have a working model with m
and c
as model parameters. We can create a line for the data points using the calculated slope and intercept:
- Recall that
$y = mx + c$ . We can now use slope and intercept values along with X data points (features) to calculate the Y data points (labels) of the regression line.
Write a function reg_line()
that takes in slope, intercept and X vector and calculates the regression line using
def reg_line (m, c, xs):
pass
# Uncomment below
#regression_line = reg_line(m,c,X)
# Plot data and regression line
So there we have it, our least squares regression line. This is the best fit line and does describe the data pretty well (still not perfect though).
# Your answer here
So, how might you go about actually making a prediction based on this model you just made?
Now that we have a working model with m and b as model parameters. We can fill in a value of x with these parameters to identify a corresponding value of
Let's try to find a y prediction for a new value of
x_new = 7
y_predicted = None
y_predicted
# 10.155092592592592
# Plot as above and show the predicted value
You now know how to create your own models, which is great! Next, you'll find out how to determine the accuracy of your model!
In this lesson, you learned how to perform linear regression for data that are linearly related. You first calculated the slope and intercept parameters of the regression line that best fit the data. You then used the regression line parameters to predict the value (