search

lejon / PartiallyCollapsedLDA

Implementations of various fast parallelized samplers for LDA, including Partially Collapsed LDA, Light LDA, Partially Collapsed Light LDA and a very efficient Polya-Urn LDA
26 stars 20 forks source link

Test suites #12

Closed MansMeg closed 6 years ago

MansMeg commented 6 years ago

Hej Leif,

Här kommer två testsuiter baserat dels på analytiskt användande av Antoniak och dels simulering. Jag har använt följande R-kod:

m_dk <- 3
p <- rep(0,m_dk)
Psi_k <- 0.1
alpha <- 1
for(l in seq_along(p)){
  p[l] <- gamma(Psi_k*alpha)/gamma(Psi_k*alpha + m_dk) * abs(copula::Stirling1(m_dk, l)) * (Psi_k*alpha)^l
}

Djk <- c(2,2,1)
s <- 1000000
l_sum <- rep(0, s)
for(j in seq_along(Djk)){
  l_sum <- l_sum + rbinom(n = s, size = Djk[j], prob = Psi_k*alpha / (Psi_k*alpha + j - 1))
}
table(l_sum)/s