Scribd
Upload
57 views

Machine Learning (Simple Linear Regression) Using Phyton

This document discusses using simple linear regression in Python to predict salary based on years of experience. It imports necessary libraries, loads and splits a salary dataset into training and test sets. A linear regression model is fitted to the training set and used to make predictions on the test set. The results are visualized with scatter plots comparing actual and predicted salaries for both training and test sets.

Uploaded by

Anisah Andini
Copyright
© © All Rights Reserved
Available Formats
Download as DOCX, PDF, TXT or read online on Scribd
57 views

Machine Learning (Simple Linear Regression) Using Phyton

This document discusses using simple linear regression in Python to predict salary based on years of experience. It imports necessary libraries, loads and splits a salary dataset into training and test sets. A linear regression model is fitted to the training set and used to make predictions on the test set. The results are visualized with scatter plots comparing actual and predicted salaries for both training and test sets.

Uploaded by

Anisah Andini
Copyright
© © All Rights Reserved
Available Formats
Download as DOCX, PDF, TXT or read online on Scribd
You are on page 1/ 2

Machine Learning (Simple Linear Regression) using phyton

# Importing the libraries


import numpy as np
import matplotlib.pyplot as plt
import pandas as pd

# Importing the dataset


dataset = pd.read_csv('Salary.csv') #baca dataset salary
X = dataset.iloc[:, :-1].values
y = dataset.iloc[:, 1].values

# Splitting the dataset into the training set and test set
from sklearn.cross_validation import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 1/3, random_state = 0)

# Fitting Simple Linear Regression terhadap Training set


from sklearn.linear_model import LinearRegression
regressor = LinearRegression()
regressor.fit(X_train, y_train)

# Predicting the test set


y_pred = regressor.predict(X_test)

# Visualising the training set results


plt.scatter(X_train, y_train, color = 'red')
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()

# Visualising the test set results


plt.scatter(X_test, y_test, color = 'red')
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()

You might also like