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 merupakan bagian dari kursus
Model Campuran di R
Instruksi latihan
Buat fungsi expectation dengan melengkapi kode contoh. Perhatikan bahwa kini kita mempertimbangkan simpangan baku setiap klaster sebagai parameter keempatnya.
Latihan interaktif langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
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)
}