Funkce expectation
Zatím jsi se naučil/a, jak se algoritmus Expectation-Maximization používá k odhadování parametrů dvou Gaussových rozdělení, kde obě sd jsou rovny 1. Cílem tohoto cvičení je vytvořit funkci expectation, která zobecňuje krok odhadování pravděpodobností v situaci, kdy známe means, proportions a sds.
Toto cvičení je součástí kurzu
Mixture Models v R
Pokyny k cvičení
Dokonči ukázkový kód a vytvoř funkci expectation. Všimni si, že nyní pracujeme se směrodatnými odchylkami každého shluku jako se čtvrtým parametrem.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
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)
}