Merge Predictions with whole data set - Printable Version +- Python Forum (https://python-forum.io) +-- Forum: Python Coding (https://python-forum.io/forum-7.html) +--- Forum: Data Science (https://python-forum.io/forum-44.html) +--- Thread: Merge Predictions with whole data set (/thread-3840.html) |
Merge Predictions with whole data set - mayanksrivastava - Jun-29-2017 A very basic question. We have dataset with experience and salary now i wanted to merge my predictions and create an output with experience and the predicted salary, How do I do this in python. below is my code. what should i change here ? #Importing Libraries import pandas as pd import numpy as np import matplotlib.pyplot as mp #Import Dataset data = pd.read_csv('H:\Data\Simple_Linear_Regression\Salary_Data.csv') x = data.iloc[:, :-1].values y = data.iloc[:, 1:2] #Split into train and test from sklearn.cross_validation import train_test_split xtrain, xtest, ytrain, ytest = train_test_split(x,y, test_size = 1/3, random_state = 0) #Fit Linear Regression Model from sklearn.linear_model import LinearRegression regressor = LinearRegression() regressor.fit(xtrain,ytrain) regressor.score(xtrain,ytrain) #Predict Salary ypred = regressor.predict(xtest)DataSet: YearsExperience Salary 1.1 39343.0 1.3 46205.0 1.5 37731.0 2.0 43525.0 2.2 39891.0 2.9 56642.0 3.0 60150.0 3.2 54445.0 3.2 64445.0 |