Last active
June 4, 2021 06:07
-
-
Save omairaasim/30cd1de4b7fff31377304d4a6c24d545 to your computer and use it in GitHub Desktop.
Project 1
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env python3 | |
# -*- coding: utf-8 -*- | |
""" | |
Created on Sun Sep 1 19:14:35 2019 | |
@author: omairaasim | |
""" | |
# Step 1 Load Data | |
import pandas as pd | |
dataset = pd.read_csv('Salary_Data.csv') | |
X = dataset.iloc[:, :-1].values | |
y = dataset.iloc[:,1].values | |
# Step 2: Split data into training and testing | |
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) | |
# Step 3: Fit Simple Linear Regression to Training Data | |
from sklearn.linear_model import LinearRegression | |
regressor = LinearRegression() | |
regressor.fit(X_train, y_train) | |
# Step 4: Make Prediction | |
y_pred = regressor.predict(X_test) | |
# Step 5 - Visualize training set results | |
import matplotlib.pyplot as plt | |
# plot the actual data points of training set | |
plt.scatter(X_train, y_train, color = 'red') | |
# plot the regression line | |
plt.plot(X_train, regressor.predict(X_train), color='blue') | |
plt.title('Salary vs Experience (Training set)') | |
plt.xlabel('Years of Experience') | |
plt.ylabel('Salary') | |
plt.show() | |
# Step 6 - Visualize test set results | |
import matplotlib.pyplot as plt | |
# plot the actual data points of test set | |
plt.scatter(X_test, y_test, color = 'red') | |
# plot the regression line (same as above) | |
plt.plot(X_train, regressor.predict(X_train), color='blue') | |
plt.title('Salary vs Experience (Test set)') | |
plt.xlabel('Years of Experience') | |
plt.ylabel('Salary') | |
plt.show() | |
# Step 7 - Make new prediction | |
new_salary_pred = regressor.predict([[15]]) | |
print('The predicted salary of a person with 15 years experience is ',new_salary_pred) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment