fix predict - add entropy to exponent

tiemvanderdeure · Dec 15, 2023 · 1d2d52f · 1d2d52f
1 parent abed52c
commit 1d2d52f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/predict.jl b/src/predict.jl
@@ -9,7 +9,7 @@ function predict(m::MaxnetModel, x; link = CloglogLink())
         feature_cols(continuous_predictors, categorical_predictors, fe, 10)
     end
 
-    exponent = mm * m.coefs .+ m.alpha
+    exponent = mm * m.coefs .+ m.alpha .+ m.entropy
 
     GLM.linkinv.(Ref(link), exponent)
 end