|
| 1 | +library('rstan'); |
| 2 | +NMax <- 1000000; |
| 3 | +fit <- 0; |
| 4 | +Nchains <- 1; |
| 5 | +Niter <- 200; |
| 6 | +Js <- c(10,100,1000,10000); |
| 7 | +Ks <- c(10,100,1000); |
| 8 | +times <- array(NA,c(length(Js),length(Ks))); |
| 9 | +for (jidx in 1:length(Js)) { |
| 10 | + J <- Js[jidx]; |
| 11 | + for (kidx in 1:length(Ks)) { |
| 12 | + K <- Ks[kidx]; |
| 13 | + if (Js[jidx] * Ks[kidx] <= NMax) { |
| 14 | + source('irt_sim.R'); |
| 15 | + if (jidx == 1 && kidx == 1) { |
| 16 | + fit <- stan(file='irt.stan', |
| 17 | + data=list(J=J,K=K,N=N,jj=jj,kk=kk,y=y), |
| 18 | + init=0, seed = 23, |
| 19 | + iter=2, |
| 20 | + chains=1) |
| 21 | + } |
| 22 | + t_start <- proc.time()[3]; |
| 23 | + print(paste("J=",J,", K=",K," N=",J*K,sep=""),quote=F); |
| 24 | + fit <- stan(file='irt.stan', |
| 25 | + data=list(J=J,K=K,N=N,jj=jj,kk=kk,y=y), |
| 26 | + fit=fit, |
| 27 | + # init=0, |
| 28 | + iter=Niter, |
| 29 | + chains=Nchains, |
| 30 | + seed=23); |
| 31 | + t_end <- proc.time()[3]; |
| 32 | + t_elapsed <- t_end - t_start; |
| 33 | + times[jidx,kidx] <- t_elapsed / Nchains / (Niter/2); |
| 34 | + } |
| 35 | + } |
| 36 | +} |
| 37 | + |
| 38 | +print("raters,items,N,sec/iter",quote=F); |
| 39 | +for (jidx in 1:length(Js)) { |
| 40 | + for (kidx in 1:length(Ks)) { |
| 41 | + if (Js[jidx] * Ks[kidx] <= NMax) { |
| 42 | + print(paste(Js[jidx],Ks[kidx],Js[jidx]*Ks[kidx],times[jidx,kidx], |
| 43 | + sep=","), |
| 44 | + quote=F); |
| 45 | + } |
| 46 | + } |
| 47 | +} |
0 commit comments