got negbinom going for unstd series

quantifish · Oct 3, 2023 · 7b7147f · 7b7147f
1 parent 64cd0d3
commit 7b7147f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/R/get-index.R b/R/get-index.R
@@ -37,7 +37,7 @@ get_unstandarsied <- function(fit, year = "year", rescale = 1) {
 
   if (!is.brmsfit(fit)) stop("fit is not an object of class brmsfit.")
 
-  if (fit$family$family == "bernoulli" | grepl("hurdle", fit$family$family)) {
+  if (fit$family$family %in% c("bernoulli", "negbinomial") | grepl("hurdle", fit$family$family)) {
     prop <- data.frame(y = fit$data[,1], Year = fit$data[,year]) %>%
       mutate(y = ifelse(.data$y > 0, 1, 0)) %>%
       group_by(.data$Year) %>%
@@ -57,8 +57,8 @@ get_unstandarsied <- function(fit, year = "year", rescale = 1) {
   gm <- geo_mean(unstd$cpue)
 
   fout <- unstd %>%
-    mutate(Mean = .data$cpue, Median = .data$cpue) %>%
-    select(-.data$cpue)
+    mutate(Mean = cpue, Median = cpue) %>%
+    select(-cpue)
 
   # Rescale the series
   if (rescale == "raw") {