Kom igångKom igång gratis

Skapa ett särdrag manuellt

Efter lite efterforskningar tillsammans med ditt team kommer du ihåg att gravitationskraften mellan två kroppar följer Newtons formel:

$$F = G\frac{m_1m_2}{r^2}$$.

Du kan inte använda formeln direkt eftersom massorna är okända, men du kan anpassa en regressionsmodell för force som en funktion av inv_square_distance. Den utökade datamängden df som du skapade i föregående övning har laddats in åt dig.

Den här övningen är en del av kursen

Feature Engineering i R

Visa kurs

Övningsinstruktioner

  • Skapa en ny variabel inv_square_distance definierad som det inverterade kvadratavståndet och lägg till den i dataramen df.
  • Bygg en enkel regressionsmodell med lm() för force mot inv_square_distance och spara den som lr_force_2.
  • Lägg till dina prediktioner till df_inverse med bind_cols.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# 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()
Redigera och kör kod