From 1d2d52f7706ebddd87bbdcada13b48609e4f2fd3 Mon Sep 17 00:00:00 2001 From: tiemvanderdeure Date: Fri, 15 Dec 2023 11:43:23 +0100 Subject: [PATCH] fix predict - add entropy to exponent --- src/predict.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/predict.jl b/src/predict.jl index ecdcf6b..915f671 100644 --- 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 \ No newline at end of file