Fungsi expectation
Sejauh ini, Anda telah mempelajari bagaimana algoritma Expectation-Maximization digunakan untuk mengestimasi parameter dua distribusi Gaussian dengan sd keduanya sama dengan 1. Tujuan latihan ini adalah membuat fungsi expectation, yang menggeneralisasi langkah estimasi peluang ketika kita mengetahui means, proportions, dan sds.
Latihan ini adalah bagian dari kursus
Model Campuran di R
Petunjuk latihan
Buat fungsi expectation dengan melengkapi kode contoh. Perhatikan bahwa kini kita mempertimbangkan simpangan baku setiap klaster sebagai parameter keempatnya.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
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)
}