Machine Learning A-Z Simple Linear Regression

Machine Learning A-Z

Simple Linear Regression

#Importing the libraries
import tensorflow as tf
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

# feature scaling
'''from sklearn.preprocessing import StandardScaler
sc_X = StandardScaler()
X_train = sc_X.transform(X_test)
sc_y = StandardScaler()
y_train = sc_y.fit_transform(y_train)'''

#fitting simple linear regression to 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)

#visualising the training set results
plt.scatter(X_train,y_train,color ='green')
plt.plot(X_train,regressor.predict(X_train),color = 'red')
plt.title('Salary VS Experience(training set)')
plt.xlabel('Years of Experience')
plt.show()

#visualising the test set results
plt.scatter(X_test,y_test,color ='blue')
plt.plot(X_train,regressor.predict(X_train),color = 'red')
plt.title('Salary VS Experience(test set)')
plt.xlabel('Years of Experience')
plt.show()