Skip to content

Instantly share code, notes, and snippets.

@josealvarez97
Created September 24, 2018 19:07
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save josealvarez97/4a22253debbdc0629d2d87aaf38ba229 to your computer and use it in GitHub Desktop.
Save josealvarez97/4a22253debbdc0629d2d87aaf38ba229 to your computer and use it in GitHub Desktop.
# Produce model from training set
LinearRegression.1 <- lm(clickperview ~ timeofday + Post_type + promotion + target_audience, data=Lironne_DataSet_Training)
summary(LinearRegression.1)
predict(LinearRegression.1,Lironne_DataSet_Training) == Lironne_DataSet_Training
# Calculate RMSE for training set
RMSE.trainingset <- sqrt(mean( (Lironne_DataSet_Training$clickperview - predict(LinearRegression.1, Lironne_DataSet_Training))^2 ))
RMSE.trainingset
# Calculate RMSE for test set
RMSE.testset <- sqrt(mean( (Lironne_DataSet_Test$clickperview - predict(LinearRegression.1, Lironne_DataSet_Test))^2 ))
RMSE.testset
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment