Función de maximización
Vimos que el algoritmo EM es un método iterativo con dos pasos: expectativa y maximización. En el ejercicio anterior creaste la función expectation. Ahora, crea la función maximization, que toma el data frame con las probabilidades y devuelve las estimaciones de las medias y las proporciones.
Este ejercicio forma parte del curso
Modelos de mezcla en R
Instrucciones del ejercicio
Crea la función maximization completando el código de ejemplo.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
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)
}