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

Функция ожидания

К этому моменту вы узнали, как алгоритм EM используется для оценки параметров двух гауссовых распределений, у которых оба значения sd равны 1. Цель этого упражнения — создать функцию expectation, которая обобщает шаг оценки вероятностей при известных means, proportions и sds.

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

Смесевые модели в R

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

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

Создайте функцию expectation, дополнив предложенный код. Обратите внимание, что теперь мы учитываем стандартные отклонения каждого кластера в качестве четвёртого параметра.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

expectation <- ___(___, means, proportions, ___){
  # Estimate the probabilities
  exp_data <- data %>% 
    mutate(prob_from_cluster1 = ___[1] * dnorm(x, mean = means[1], sd = ___[1]),
           prob_from_cluster2 = ___[2] * dnorm(x, mean = means[2], sd = ___[2]),
           prob_cluster1 = prob_from_cluster1/(prob_from_cluster1 + prob_from_cluster2),
           prob_cluster2 = prob_from_cluster2/(prob_from_cluster1 + prob_from_cluster2)) %>% 
    select(x, ___, ___)
    
  # Return data with probabilities
  return(exp_data)
}
Редактировать и запускать код