Salary Prediction of Employees in Machine Learning

Program 1

Employee Dataset

# ML Model for Salary Predication

import pandas as pd
import numpy as np
import matplotlib.pyplot as mlt
from sklearn import  linear_model
mydf=pd.read_excel("D://MLFile/employee.xlsx")
print(mydf)
mymodel=linear_model.LinearRegression()
mymodel.fit(mydf.drop('salary',axis='columns'),mydf.salary)
ex=float(input("Enter an Experience: "))
print("Predicated Salary: ",mymodel.predict([[ex]]))
alpha=mymodel.intercept_
beta=mymodel.coef_
print("Formula Predication: ",(alpha+beta*ex))
mydf_exp=pd.read_excel("D://MLFile/empexperience.xlsx")
print(mydf_exp)
predsal=mymodel.predict((mydf_exp))
print("-------------------------------------------------")
mydf_exp['Salary']=predsal
print(mydf_exp)
mydf_exp.to_excel("D://MLFile/newempData.xlsx")
mlt.scatter(mydf_exp.experience,mydf_exp.Salary,color="green",marker="+")
mlt.plot(mydf_exp.experience,mydf_exp.Salary,color="red")
mlt.xlabel("Years of Experience")
mlt.ylabel("Salary(In Rupees)")
mlt.show()
print("Predication Done Successfully............")