НачатьНачать бесплатно

Создание признака вручную

После совместного исследования с командой вы вспоминаете, что сила гравитационного притяжения между двумя телами подчиняется формуле Ньютона:

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

Использовать формулу напрямую не получится, так как массы неизвестны, однако можно построить регрессионную модель зависимости force от inv_square_distance. Дополненный набор данных df, созданный в предыдущем упражнении, уже загружен для вас.

Это упражнение является частью курса

Конструирование признаков в R

Посмотреть курс

Инструкции к упражнению

  • Создайте новую переменную inv_square_distance, определив её как обратную величину квадрата расстояния, и добавьте её в датафрейм df.
  • Постройте простую регрессионную модель с помощью lm(), где force является зависимой переменной, а inv_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()
Редактировать и запускать код