Machine Learning

How to run Linear regression in Python scikit-Learn

You know that linear regression is a popular technique and you might as well seen the mathematical equation of linear regression. But do you know how to implement a linear regression in Python?? If so don’t read this post because this post is all about implementing linear regression in Python. There are several ways in which you can do that, you can do linear regression using numpy, scipy, stats model and sckit learn. But in this post I am going to use scikit learn to perform linear regression.
Scikit-learn is a powerful Python module for machine learning. It contains function for regression, classification, clustering, model selection and dimensionality reduction. Today, I will explore the sklearn.linear_model module which contains “methods intended for regression in which the target value is expected to be a linear combination of the input variables”.
In this post, I will use Boston Housing data set, the data set  contains information about the housing values in suburbs of Boston. This dataset was originally taken from the StatLib library which is maintained at Carnegie Mellon University and is now available on the UCI Machine Learning Repository.  UCI machine learning repository contains many interesting data sets, I encourage you to go through it.
So come on lets have fun with linear regression,
Exploring Boston Housing Data Set 
The first step is to import the required Python libraries into Ipython Notebook.
Explore 1
This data set is available in  sklearn Python module, so I will access it using scikitlearn. I am going to import Boston data set into Ipython notebook and store it in a variable called boston.
The object boston is a dictionary, so you can explore the keys of this dictionary.
boston keys
boston data shape
I am going to print the feature names of boston data set.
boston features
I will see the description of this data set to know more about it. In this  data set I have 506 instances(rows) and 13 attributes or parameters(columns). The goal of this exercise is to predict the  housing prices in boston region using the features given.
boston description
I am going to convert into a pandas data frame.
Pandas DataFrame
As you can see the column names are just numbers, so I am going to replace those numbers with the feature names.
bos columns contains the housing prices.
Boston target
I am going to add these target prices to the bos data frame.
Bos Price
Scikit Learn
In this section I am going to  fit a linear regression model and predict the Boston housing prices. I will use the least squares method as the way to estimate the coefficients.
Y = boston housing price(also called “target” data in Python)
X = all the other features (or independent variables)
First, I am going to import linear regression from sci-kit learn module. Then I am going to drop the price column as I want only the parameters as my X values.  I am going to store linear regression object in a variable called lm.
Skitlearn linear model
If you want to look inside the linear regression object, you can do so by typing LinearRegression. and the press <tab> key. This will give a list of functions available inside linear regression object.
Important functions to keep in mind while fitting a linear regression model are: -> fits a linear model
lm.predict() -> Predict Y using the linear model with estimated coefficients
lm.score() -> Returns the coefficient of determination (R^2). A measure of how well observed outcomes are replicated by the model, as the proportion of total variation of outcomes explained by the model. 
You can also explore the functions inside lm object by pressing lm.<tab>
.coef_  gives the coefficients and .intercept_  gives the estimated intercepts.
Fitting a Linear Model
I am going to use all 13 parameters to fit a linear regression  model.  Two other parameters that you can pass to linear regression object are fit_intercept and normalize.
In [20]:, bos.PRICE)
Out[20]: LinearRegression(copy_X=True, fit_intercept=True, normalize=False)
I am going to print the intercept and number of coefficients.
Estimated Coeff
I then construct a data frame that contains features and estimated coefficients.
pd data frame
As you can see from the data frame that there is a high correlation between RM and prices. Lets plot a scatter plot between True housing prices and True RM.
Scatter plot
Relationship between RM and Price
As you can see that there is a positive correlation between RM and housing prices.
Predicting Prices
I am going to calculate the predicted prices (Y^i)  using lm.predict. Then I display the first 5 housing prices. These are my predicted housing prices.
lm predict
Then I plot a scatter plot to compare true prices and the predicted prices.
Scatter plot in the pandas
Prices vs predicted prices
You can notice that there is some error in  the prediction as the housing prices increase.
Lets calculate the mean squared error.
MSE full But if you fit linear regression for  one feature the error will be very high. Lets take the feature ‘PTRATIO’ and calculate the mean squared error.
Linear regression and fitting
MSE prat
The mean squared error has increased. So this shows that a single feature is not a good predictor of housing prices.
Training and validation data sets
In practice you wont implement linear regression on the entire data set, you will have to split the data sets into training and test data sets. So that you train your model on training data and see how well it performed on test data.
How not to do train-test split:
train-test split
You can create training and test data sets manually, but this is not the right way to do, because you may be training your model on less expensive houses and testing on expensive houses.
How to do train-test split:
You have to divide your data sets randomly. Scikit learn provides a function called train_test_split to do this.
Xtrain and Xtest
I am going to build a linear regression model using my  train-test data sets.
Linear reg
Then I calculate the mean squared error for training and test data.
print “Fit a model X_train, and calculate MSE with Y_train:”, np.mean((Y_train – lm.predict(X_train)) ** 2)
print “Fit a model X_train, and calculate MSE with X_test, Y_test:”, np.mean((Y_test – lm.predict(X_test)) ** 2)
Fit a model X_train, and calculate MSE with Y_train: 19.5467584735 Fit a model X_train, and calculate MSE with X_test, Y_test: 28.5413672756
Residual Plots
Residual plots are a good way to visualize the errors in your data.  If you have done a good job then your data should be randomly scattered around line zero. If you see structure in your data, that means your model is not capturing some thing. Maye be there is a interaction between 2 variables that you are not considering, or may be you are measuring time dependent data. If you get some structure in your data, you should go back to your model and check whether you are doing a good job with your parameters.
Plt scatter
Residual plot
To recap what I have done till now,

  1. I explored the boston data set and then renamed its column names.
  2. I explored the boston data set using .DESCR, my goal was to predict the housing prices  using the given features.
  3. I used Scikit learn to fit linear regression to the entire data set and calculated the mean squared error.
  4. I made a train-test split and calculated the mean squared error for my training data and test data.
  5.  I then plotted the residuals for my training and test datasets.
  1. Yael fregier 2 years ago

    Hi, I have read some of the entries of your former blog “big data examiner”, and I was very surprised by seeing that it is no longer available online. It is a great ressource, please, try to get it back to life!

  2. Puneet Mathur 2 years ago

    Your code gives error on the MSE code block
    print( np.mean((Y_test – lm.predict(X_test)) ** 2))
    Traceback (most recent call last):
    File “”, line 1, in
    print( np.mean((Y_test – lm.predict(X_test)) ** 2))
    raise ValueError(“input must be a square array”)
    ValueError: input must be a square array

  3. Madhav 2 years ago

    Simplified and upto the point.

  4. Mike 1 year ago

    Great article, thanks!

  5. NunoAndre 1 year ago

    You can do simply

    from scipy import stats

    instead of

    import scipy.stats as stats

  6. neeraj 12 months ago

    what the meaning of the mean square error with training and test data , with train data MSE is 19 and with test data MSE is 28

    • Shripad 4 months ago

      because MSE is inversely proportional to sample size

  7. Ciaran Duncan 11 months ago

    Great Article, Really helped me

  8. Abdellah Madani 9 months ago

    Thanks, great

  9. Zachery Poche 9 months ago

    Excellent read. Thanks 🙂

  10. Suryakant Soni 8 months ago

    fit function giving error..

    • shripad 4 months ago

      what error it is giving?

  11. reding1234 6 months ago

    great resource and information

  12. Hetal Rajpura 6 months ago

    Thank you very much

  13. Victoria M Ng 6 months ago

    Thank you! This was extremely helpful!

  14. Inho Cho 6 months ago

    Thank you very much…

  15. Shripad Bhat 4 months ago

    Good article. Informative. Thank you.
    I have updated where readers may encounter error in How to do train-test split due to update issue:
    from sklearn.model_selection import train_test_split
    X_train, X_test, Y_train, Y_test = train_test_split( X, bos.PRICE, test_size=0.33, random_state=5)
    print (X_train.shape)
    print (X_test.shape)
    print (Y_train.shape)
    print (Y_test.shape)

Leave a Comment

Your email address will not be published.

You may also like

Crayon Yoda

Pin It on Pinterest