A tentative model
You are handed a data set with measures of the gravitational force between two bodies at different distances and are challenged to build a simple model to predict such force given a specific distance. Initially, you want to stick to simple linear regression. The data consist of 120 pairs of distance
and force
, and is loaded for you as newton
.
This exercise is part of the course
Feature Engineering in R
Exercise instructions
- Build a linear model for the
newton
data using the linear model from baseR
function and assign it tolr_force
. - Create a new data frame
df
by binding the prediction values to the originalnewton
data. - Generate a scatterplot of
force
versusdistance
usingggplot()
. - Add a regression line to the scatterplot with the fitted values.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Build a linear model for the newton the data and assign it to lr_force
lr_force <- ___(force ~ distance, data = ___)
# Create a new data frame by binding the prediction values to the original data
df <- newton %>% ___(lr_pred = predict(lr_force))
# Generate a scatterplot of force vs. distance
df %>%
ggplot(aes(x = distance, y = force)) +
geom____() +
# Add a regression line with the fitted values
geom_line(aes(y = ___), color = "blue", lwd = .75) +
ggtitle("Linear regression of force vs. distance") +
theme_classic()