In-sample RMSE for linear regression on diamonds
As you saw in the video, included in the course is the diamonds dataset, which is a classic dataset from the ggplot2 package. The dataset contains physical attributes of diamonds as well as the price they sold for. One interesting modeling challenge is predicting diamond price based on their attributes using something like a linear regression.
Recall that to fit a linear regression, you use the lm() function in the following format:
mod <- lm(y ~ x, my_data)
To make predictions using mod on the original data, you call the predict() function:
pred <- predict(mod, my_data)
Latihan ini adalah bagian dari kursus
Machine Learning with caret in R
Petunjuk latihan
- Fit a linear model on the
diamondsdataset predictingpriceusing all other variables as predictors (i.e.price ~ .). Save the result tomodel. - Make predictions using
modelon the full original dataset and save the result top. - Compute errors using the formula \(errors = predicted - actual\). Save the result to
error. - Compute RMSE using the formula you learned in the video and print it to the console.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Fit lm model: model
# Predict on full data: p
# Compute errors: error
# Calculate RMSE