開始使用免費開始

手動打造一個特徵

在和團隊做了一些研究後,你回想起兩個物體之間的萬有引力符合牛頓公式:

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

你無法直接使用此公式,因為質量未知。不過,你可以將 forceinv_square_distance 做迴歸。上一題你已建立並擴充的資料集 df 已為你載入。

本練習屬於課程

R 的特徵工程

檢視課程

練習說明

  • 建立新變數 inv_square_distance,其定義為距離平方的倒數,並將它加入資料框 df
  • 使用 lm() 建立一個簡單迴歸模型,將 forceinv_square_distance,並將模型儲存為 lr_force_2
  • 把你的預測結果與 df_inverse 進行欄位合併。

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# 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()
編輯並執行程式碼