Advertisement
Guest User

Untitled

a guest
Sep 28th, 2015
152
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.68 KB | None | 0 0
  1.  
  2. require(rstan)
  3. x0=c(3,1,2,5); y0=table(x0)
  4. x=as.numeric(names(y0)); y=unname(y0)
  5. dat=list(x=x, y=y, Ndata=length(x0), Nunique=length(x))
  6. ####
  7. model <- '
  8. data {
  9. int<lower=0> Ndata; int<lower=0> Nunique;
  10. int<lower=0> y[Ndata]; int<lower=0> x[Ndata];
  11. }
  12. parameters {
  13. real<lower=0, upper=1> theta;
  14. }
  15. model {
  16. for(i in 1:Nunique){
  17. y[i] ~ poisson(Ndata*(theta^(x[i]-1) - theta^x[i]));
  18. }
  19. }
  20. '
  21. ####
  22. fit <- stan(model_code = model, data = dat, iter = 100000, chains = 3, warmup=500)
  23. fit2<-extract(fit)
  24. hist(fit2$theta, breaks=seq(0,1,by=.01))
  25. mean(fit2$theta)
  26. median(fit2$theta)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement