Crearea manuală a unei caracteristici
În urma unor cercetări făcute împreună cu echipa ta, îți amintești că forța gravitațională de atracție dintre două corpuri respectă formula lui Newton:
$$F = G\frac{m_1m_2}{r^2}$$.
Nu poți folosi formula direct, deoarece masele sunt necunoscute, însă poți antrena un model de regresie al force în funcție de inv_square_distance. Setul de date augmentat df, pe care l-ai construit în exercițiul anterior, a fost încărcat pentru tine.
Acest exercițiu face parte din cursul
Ingineria caracteristicilor în R
Instrucțiuni pentru exercițiu
- Creează o nouă variabilă
inv_square_distance, definită ca inversul distanței la pătrat, și adaug-o la data frame-uldf. - Construiește un model de regresie simplă folosind
lm()alforcefață deinv_square_distanceși salvează-l calr_force_2. - Adaugă predicțiile la
df_inversefolosindbind_cols.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# Create a new variable inv_square_distance
df_inverse <- df %>% ___(inv_square_distance = 1/distance^2)
# Build a simple regression model
lr_force_2 <- lm(force ~ ___, data = df_inverse)
# Bind your predictions to df_inverse
df_inverse <- df_inverse %>% ___(lr2_pred = predict(lr_force_2))
df_inverse %>% ggplot(aes(x = distance, y = force)) +
geom_point() +
geom_line(aes(y = lr2_pred), col = "blue", lwd = .75) +
ggtitle("Linear regression of force vs. inv_square_distance") +
theme_classic()