max/ourMELONS
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Improved array dropping

Browse source
This commit is contained in:
Waldir Leoncio 2021-02-01 10:01:02 +01:00
parent edd0cac775
commit aa8a066530
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
R/computePopulationLogml.R
View file

@ -13,14 +13,14 @@ computePopulationLogml <- function(pops, adjprior, priorTerm = 0) {
# ======================================================== #
# Computation #
# ======================================================== #
isarray <- length(dim(repmat(adjprior, c(1, 1, length(pops))))) > 2
term1 <- squeeze(
# FIXME: assumes COUNTS has 3 dims. Where does this come from?
sum(
sum(
reshape(
lgamma(
repmat(adjprior, c(1, 1, length(pops))) +
COUNTS[seq_len(nrow(adjprior)), seq_len(ncol(adjprior)), pops, drop=FALSE]
COUNTS[seq_len(nrow(adjprior)), seq_len(ncol(adjprior)), pops, drop=!isarray]
),
c(x, y, z)
),