Функція максимізації
Ми бачили, що алгоритм EM — це ітеративний метод із двома кроками: очікування та максимізація. У попередній вправі ви створили функцію expectation. Тепер створіть функцію maximization, яка приймає датафрейм з ймовірностями і повертає оцінки середніх та часток.
Ця вправа є частиною курсу
Мішанинні моделі в R
Інструкції до вправи
Створіть функцію maximization, доповнивши зразок коду.
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
maximization <- function(___){
means_estimates <- data_with_probs %>%
summarise(mean_1 = sum(x * ___) / ___(prob_cluster1),
mean_2 = sum(x * ___) / ___(prob_cluster2)) %>%
as.numeric()
props_estimates <- data_with_probs %>%
summarise(proportion_1 = ___(prob_cluster1),
proportion_2 = 1 - ___) %>%
as.numeric()
list(means_estimates, props_estimates)
}