more supported models

vincentarelbundock · Sep 11, 2023 · 75e7999 · 75e7999
1 parent b871e06
commit 75e7999
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/R/sanitize_type.R b/R/sanitize_type.R
@@ -13,6 +13,14 @@ sanitize_type <- function(model, type, calling_function = NULL) {
     if (calling_function %in% c("slopes", "comparisons")) {
         dict <- dict[dict$type != "linkinv(link)", , drop = FALSE]
     }
+
+    # fixest: linkinv(link) only supported for glm model
+    if (inherits(model, "fixest")) {
+        if (!isTRUE(hush(model[["method_type"]]) %in% c("feglm"))) {
+            dict <- dict[dict$type != "linkinv(link)", , drop = FALSE]
+        }
+    }
+
     dict <- dict[dict$class == cl, , drop = FALSE]
     checkmate::assert_choice(type, choices = dict$type, null.ok = TRUE)
     if (is.null(type)) {

diff --git a/R/type_dictionary.R b/R/type_dictionary.R
@@ -49,6 +49,7 @@ hxlr,scale
 hxlr,density
 ivpml,pr
 ivpml,xb
+fixest,linkinv(link)
 fixest,response
 fixest,link
 hurdle,response
@@ -59,6 +60,7 @@ iv_robust,response
 lm,response
 gam,response
 gam,link
+Gam,linkinv(link)
 Gam,response
 Gam,link
 geeglm,response
@@ -102,6 +104,7 @@ mhurdle,E
 mhurdle,Ep
 mhurdle,p
 mlogit,response
+negbin,linkinv(link)
 negbin,response
 negbin,link
 ols,lp