source("gp-mix-em.r") d <- read.table ("a3data2.txt", head=FALSE) X <- as.matrix(d) Xtrn <- X[1:200,] Xtst <- X[-(1:200),] pdf("a3plots2.pdf",height=9,width=6,pointsize=9) par(mfrow=c(4,3)) for (K in c(10,15)) { for (s in c(0.001,0.6)) { for (seed in 1:3) { set.seed(seed) cat("\nRUNNING EM WITH K =",K,", s =",s,", seed =",seed,"\n\n") fit <- gp_mix_em (Xtrn, K, s, 1.0, 200, trace=FALSE) cat("sigma:\n"); print(round(fit$sigma,4)) cat("pi:\n"); print(round(fit$mpi,4)) cat("mu:\n"); print(round(fit$mu,4)) lpd <- mix_pred (fit$mpi, fit$mu, fit$sigma, Xtst) plot(Xtrn[,1],Xtrn[,2],pch=20) points(fit$mu[,1],fit$mu[,2],pch=20,type="b",cex=2,col="red") title(paste("K:",K," s:",s," seed:",seed," lpd:",round(lpd,3))) } } } dev.off()