Категориальные объясняющие переменные

Введение в регрессию на R

Richie Cotton

Data Evangelist at DataCamp

Набор данных о рыбах

  • Каждая строка соответствует одной рыбе.
  • В наборе данных 128 строк.
  • Представлено 4 вида рыб.
species mass_g
Bream 242.0
Perch 5.9
Pike 200.0
Roach 40.0
... ...
Введение в регрессию на R

Визуализация 1 числовой и 1 категориальной переменной

library(ggplot2)

ggplot(fish, aes(mass_g)) +
  geom_histogram(bins = 9) +
  facet_wrap(vars(species))

Фасетная гистограмма количества рыб по их массе. Каждая панель соответствует виду: лещ, окунь, щука или плотва.

Введение в регрессию на R

Описательная статистика: средняя масса по видам

fish %>% 
  group_by(species) %>% 
  summarize(mean_mass_g = mean(mass_g))

# A tibble: 4 x 2
  species mean_mass_g
  <chr>         <dbl>
1 Bream          618.
2 Perch          382.
3 Pike           719.
4 Roach          152.
Введение в регрессию на R

Линейная регрессия

lm(mass_g ~ species, data = fish)

Call:
lm(formula = mass_g ~ species, data = fish)

Coefficients:
 (Intercept)  speciesPerch   speciesPike  speciesRoach  
       617.8        -235.6         100.9        -465.8
Введение в регрессию на R

Без свободного члена

lm(mass_g ~ species + 0, data = fish)

Call:
lm(formula = mass_g ~ species + 0, data = fish)

Coefficients:
speciesBream  speciesPerch   speciesPike  speciesRoach  
       617.8         382.2         718.7         152.0
Введение в регрессию на R

Давайте потренируемся!

Введение в регрессию на R

Preparing Video For Download...