ПочатиПочніть безкоштовно

Функція expectation

Дотепер ви дізналися, як алгоритм Expectation–Maximization використовується для оцінювання параметрів двох гаусових розподілів із 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)
}
Редагувати та запускати код