fix default link in mlj predict

tiemvanderdeure · Dec 23, 2023 · 3a0b72a · 3a0b72a
1 parent 1c073fe
commit 3a0b72a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/mlj_interface.jl b/src/mlj_interface.jl
@@ -82,7 +82,7 @@ function MMI.fit(m::MaxnetBinaryClassifier, verbosity::Int, X, y)
 end
 
 function MMI.predict(m::MaxnetBinaryClassifier, (fitresult, decode), Xnew; 
-    link = Cloglog(), clamp = false)
+    link = CloglogLink(), clamp = false)
     p = predict(fitresult, Xnew; link = link, clamp = clamp)
     MMI.UnivariateFinite(decode, [1 .- p, p])
 end