- Importing libraries and datasets
- Splitting the dataset
- Training the simple Linear Regression model on the Training set
- Predicting and visualizing the test set results
- Visualizing the training set results
- Making a single prediction
- Getting the final linear regression equation (with values of the coefficients)
y = bo + b1 * x1
- y: Dependent Variable (DV)
- x: InDependent Variable (IV)
- b0: Intercept Coefficient
- b1: Slope of Line Coefficient
- Using
sklearn.linear_model
,LinearRegression
model
from sklearn.linear_model import LinearRegression
#To Create Instance of Simple Linear Regression Model
regressor = LinearRegression()
#To fit the X_train and y_train
regressor.fit(X_train, y_train)
y_pred = regressor.predict(X_test)
Important note: "predict" method always expects a 2D array as the format of its inputs.
- And putting 12 into a double pair of square brackets makes the input exactly a 2D array:
regressor.predict([[12]])
print(f"Predicted Salary of Employee with 12 years of EXP: {regressor.predict([[12]])}" )
#Output: Predicted Salary of Employee with 12 years of EXP: [137605.23485427]
#Plot predicted values
plt.scatter(X_test, y_test, color = 'red', label = 'Predicted Value')
#Plot the regression line
plt.plot(X_train, regressor.predict(X_train), color = 'blue', label = 'Linear Regression')
#Label the Plot
plt.title('Salary vs Experience (Test Set)')
plt.xlabel('Years of Experience')
plt.ylabel('Salary')
#Show the plot
plt.show()
- General Formula:
y_pred = model.intercept_ + model.coef_ * x
print(f"b0 : {regressor.intercept_}")
print(f"b1 : {regressor.coef_}")
b0 : 25609.89799835482
b1 : [9332.94473799]
Linear Regression Equation: Salary = 25609 + 9332.94×YearsExperience
- compare how well different algorithms perform on a particular dataset.
- For regression algorithms, three evaluation metrics are commonly used:
- R Square/Adjusted R Square > Percentage of the output variability
- Mean Square Error(MSE)/Root Mean Square Error(RMSE) > to compare performance between different regression models
- Mean Absolute Error(MAE) > to compare performance between different regression models
- R Square measures how much of variability in predicted variable can be explained by the model.
Variance
is a measure in statistics defined as the average of the square of differences between individual point and the expected value.- R Square value: between 0 to 1 and bigger value indicates a better fit between prediction and actual value.
- However, it does not take into consideration of overfitting problem.
- If your regression model has many independent variables, because the model is too complicated, it may fit very well to the training data
- but performs badly for testing data.
- Solution: Adjusted R Square
- is introduced Since R-square can be increased by adding more number of variable and may lead to the over-fitting of the model
- Will penalise additional independent variables added to the model and adjust the metric to prevent overfitting issue.
- In Python, you can calculate R Square using
Statsmodel
orSklearn
Package
import statsmodels.api as sm
X_addC = sm.add_constant(X)
result = sm.OLS(Y, X_addC).fit()
print(result.rsquared, result.rsquared_adj)
# 0.79180307318 0.790545085707
- around 79% of dependent variability can be explain by the model and adjusted R Square is roughly the same as R Square meaning the model is quite robust
- While R Square is a relative measure of how well the model fits dependent variables
- Mean Square Error (MSE) is an absolute measure of the goodness for the fit.
- Root Mean Square Error(RMSE) is the square root of MSE.
- It is used more commonly than MSE because firstly sometimes MSE value can be too big to compare easily.
- Secondly, MSE is calculated by the square of error, and thus square root brings it back to the same level of prediction error and make it easier for interpretation.
from sklearn.metrics import mean_squared_error
import math
print(mean_squared_error(Y_test, Y_predicted))
print(math.sqrt(mean_squared_error(Y_test, Y_predicted)))
# MSE: 2017904593.23
# RMSE: 44921.092965684235
- Compare to MSE or RMSE, MAE is a more direct representation of sum of error terms.
from sklearn.metrics import mean_absolute_error
print(mean_absolute_error(Y_test, Y_predicted))
#MAE: 26745.1109986
Before choosing Linear Regression, need to consider below assumptions
- Linearity
- Homoscedasticity
- Multivariate normality
- Independence of errors
- Lack of multicollinearity
- Since
State
is categorical variable => we need to convert it intodummy variable
- No need to include all dummy variable to our Regression Model => Only omit one dummy variable
- Ho :
Null Hypothesis (Universe)
- H1 :
Alternative Hypothesis (Universe)
- For example:
- Assume
Null Hypothesis
is true (or we are living in Null Universe)
- Assume
- 5 methods of Building Models
- Throw in all variables in the dataset
- Usage:
- Prior knowledge about this problem; OR
- You have to (Company Framework required)
- Prepare for Backward Elimination
- Step 1: Select a significance level (SL) to stay in the model (e.g: SL = 0.05)
- Step 2: Fit the full model with all possible predictors
- Step 3: Consider Predictor with Highest P-value
- If P > SL, go to Step 4, otherwise go to [FIN : Your Model Is Ready]
- Step 4: Remove the predictor
- Step 5: Re-Fit model without this variable
- Step 1: Select a significance level (SL) to enter in the model (e.g: SL = 0.05)
- Step 2: Fit all simple regression models (y ~ xn). Select the one with Lowest P-value for the independent variable.
- Step 3: Keep this variable and fit all possible regression models with one extra predictor added to the one(s) you already have.
- Step 4: Consider the predicotr with Lowest P-value. If P < SL (i.e: model is good), go STEP 3 (to add 3rd variable into the model and so on with all variables we have left), otherwise go to [FIN : Keep the previous model]
- Step 1: Select a significant level to enter and to stay in the model:
e.g: SLENTER = 0.05, SLSTAY = 0.05
- Step 2: Perform the next step of Forward Selection (new variables must have: P < SLENTER to enter)
- Step 3: Perform ALL steps of Backward Elimination (old variables must have P < SLSTAY to stay) => Step 2.
- Step 4: No variables can enter and no old variables can exit => [FIN : Your Model Is Ready]
- Step 1: Select a criterion of goodness of ift (e.g Akaike criterion)
- Step 2: Construct all possible regression Models:
2^(N) - 1
total combinations, where N: total number of variables - Step 3: Select the one with best criterion => [FIN : Your Model Is Ready]
- Note: Backward Elimination is irrelevant in Python, because the Scikit-Learn library automatically takes care of selecting the statistically significant features when training the model to make accurate predictions.
#no need Feature Scaling (FS) for Multi-Regression Model: y = b0 + b1 * x1 + b2 * x2 + b3 * x3,
# since we have the coefficients (b0, b1, b2, b3) to compensate, so there is no need FS.
from sklearn.model_selection import train_test_split
# NOT have to remove manually a dummy variable column because Scikit-Learn takes care of it.
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.2, random_state = 0)
#LinearRegression will take care "Dummy variable trap" & feature selection
from sklearn.linear_model import LinearRegression
regressor = LinearRegression()
regressor.fit(X_train, y_train)
y_pred = regressor.predict(X_test)
- Since this is multiple linear regression, so can not visualize by drawing the graph
#To display the y_pred vs y_test vectors side by side
np.set_printoptions(precision=2) #To round up value to 2 decimal places
#np.concatenate((tuple of rows/columns you want to concatenate), axis = 0 for rows and 1 for columns)
#y_pred.reshape(len(y_pred),1) : to convert y_pred to column vector by using .reshape()
print(np.concatenate((y_pred.reshape(len(y_pred),1), y_test.reshape(len(y_test),1)), 1))
print(regressor.coef_)
print(regressor.intercept_)
[ 8.66e+01 -8.73e+02 7.86e+02 7.73e-01 3.29e-02 3.66e-02]
42467.52924853204
Equation: Profit = 86.6 x DummyState1 - 873 x DummyState2 + 786 x DummyState3 - 0.773 x R&D Spend + 0.0329 x Administration + 0.0366 x Marketing Spend + 42467.53