Thanks to visit codestin.com
Credit goes to www.scribd.com

Scribd
Upload
3 views1 page

Data Science: Linear Regression Basics

The document outlines the implementation of a linear regression model using diabetes patient data from sklearn. It demonstrates loading the dataset, training the model with a portion of the data, and evaluating its performance using mean squared error. Additionally, it visualizes the model's predictions against the actual test data.

Uploaded by

Atiya Falak
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
3 views1 page

Data Science: Linear Regression Basics

The document outlines the implementation of a linear regression model using diabetes patient data from sklearn. It demonstrates loading the dataset, training the model with a portion of the data, and evaluating its performance using mean squared error. Additionally, it visualizes the model's predictions against the actual test data.

Uploaded by

Atiya Falak
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

Supervised Learning : Linear Regression Model

# Diabetes patient data

import matplotlib.pyplot as plt


import numpy as np
from sklearn import datasets, linear_model
from sklearn.metrics import mean_squared_error

#Load the datasets from sklearn


diabetes = datasets.load_diabetes()

print(diabetes.keys())
#(['data', 'target', 'DESCR', 'feature_names')]
#print(diabetes.DESCR)

# Display DESCR in the one-column format index =2


diabetes_X = diabetes.data[:, np.newaxis, 2]
diabetes_X = diabetes.data[:, np.newaxis, 2]
#print(diabetes_X)

#Take 30 data to train (last 30) and test data (first 20)
diabetes_X_train = diabetes_X[:-50]
diabetes_X_test = diabetes_X[-30:]

diabetes_Y_train = diabetes.target[:-50]
diabetes_Y_test = diabetes.target[-30:]

#Model definition
model = linear_model.LinearRegression()
model.fit(diabetes_X_train,diabetes_Y_train)
diabetes_y_predicted = model.predict(diabetes_X_test)

print ("Mean squared error is :", mean_squared_error(diabetes_Y_test,


diabetes_y_predicted))

#Find weights of the regression line


print("Weights: ", model.coef_)
print("Intercept: ", model.intercept_)

#Plot the Model


plt.scatter(diabetes_X_test, diabetes_Y_test)
plt.plot(diabetes_X_test,diabetes_y_predicted)
plt.show()

You might also like