Scribd
Upload
36 views1 page

Simple Linear Regression-Codes

The document outlines a simple linear regression implementation using Python, including data import, dataset splitting, model training, and predictions. It utilizes libraries such as NumPy, Matplotlib, and Pandas, and employs Scikit-learn for model training and testing. The results are visualized with a scatter plot of the training data and the regression line.

Uploaded by

Dr Sumathy V
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as DOCX, PDF, TXT or read online on Scribd
36 views1 page

Simple Linear Regression-Codes

The document outlines a simple linear regression implementation using Python, including data import, dataset splitting, model training, and predictions. It utilizes libraries such as NumPy, Matplotlib, and Pandas, and employs Scikit-learn for model training and testing. The results are visualized with a scatter plot of the training data and the regression line.

Uploaded by

Dr Sumathy V
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as DOCX, PDF, TXT or read online on Scribd
You are on page 1/ 1

# Simple Linear Regression

# Importing the libraries


import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
# Importing the dataset
dataset = pd.read_csv('Salary_Data.csv')
X = dataset.iloc[:, :-1].values
y = dataset.iloc[:, -1].values
# Splitting the dataset into the Training set and Test set
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 1/3,
random_state = 0)
# Training the Simple Linear Regression model on the Training set
from sklearn.linear_model import LinearRegression
regressor = LinearRegression()
regressor.fit(X_train, y_train)
# Predicting the Test set results
y_pred = regressor.predict(X_test)
print(y_pred.reshape(len(y_pred),1))
regressor.intercept_
regressor.coefficint_
# Visualising the Training set results
plt.scatter(X_train, y_train, color = 'red')
plt.plot(X_train, regressor.predict(X_train), color = 'blue')

You might also like