You are on page 1of 2

Linear regression

Program

import numpy as np
import matplotlib.pyplot as plt 
import pandas as pd  
from sklearn.linear_model import LinearRegression
data = pd.read_csv('sample_dataset.csv')  
X = data.iloc[:, 0].values.reshape(-1, 1) 
Y = data.iloc[:, 1].values.reshape(-1, 1)  
linear_regressor = LinearRegression()  
linear_regressor.fit(X, Y)  
Y_pred = linear_regressor.predict(X)  
plt.scatter(X, Y)
plt.plot(X, Y_pred, color='red')
plt.title('Test Data')
plt.xlabel('SAT')
plt.ylabel('GPA')
plt.show()

Output
Linear regression

Program

import numpy as np
import matplotlib.pyplot as plt 
import pandas as pd  
from sklearn.linear_model import LinearRegression
data = pd.read_csv('sample_dataset.csv')  
X = data.iloc[:, 0].values.reshape(-1, 1) 
Y = data.iloc[:, 1].values.reshape(-1, 1)  
linear_regressor = LinearRegression()  
linear_regressor.fit(X, Y)  
Y_pred = linear_regressor.predict(X)  
plt.plot(X, Y_pred, color='red')
plt.title('Test Data')
plt.xlabel('SAT')
plt.ylabel('GPA')
plt.show()

Output

You might also like