plot.scatter(xTrain, yTrain, color = 'red')plot.plot(xTrain, linearRegressor.predict(xTrain), color = 'blue')plot.title('Salary vs Experience (Training set)')plot.xlabel('Years of Experience')plot.ylabel('Salary')plot.show()
plot.scatter(xTest, yTest, color = 'red')plot.plot(xTrain, linearRegressor.predict(xTrain), color = 'blue')plot.title('Salary vs Experience (Test set)')plot.xlabel('Years of Experience')plot.ylabel('Salary')plot.show()
linearRegressor.fit(xTrain, yTrain)