fixes #131

Author: tlienart

Project.toml

@@ -1,7 +1,7 @@
 name = "MLJLinearModels"
 uuid = "6ee0df7b-362f-4a72-a706-9e79364fb692"
 authors = ["Thibaut Lienart <tlienart@me.com>"]
-version = "0.7.1"
+version = "0.7.2"
 
 [deps]
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
@@ -35,4 +35,16 @@ StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["DelimitedFiles", "PyCall", "ForwardDiff", "Test", "Random", "RDatasets", "RCall", "MLJ", "MLJBase", "StableRNGs", "DataFrames"]
+test = [
+    "DelimitedFiles",
+    "PyCall",
+    "ForwardDiff",
+    "Test",
+    "Random",
+    "RDatasets",
+    "RCall",
+    "MLJ",
+    "MLJBase",
+    "StableRNGs",
+    "DataFrames",
+]

src/mlj/classifiers.jl

@@ -26,7 +26,7 @@ $(example_docstring("LogisticClassifier", nclasses = 2))
 @with_kw_noshow mutable struct LogisticClassifier <: MMI.Probabilistic
     "strength of the regulariser if `penalty` is `:l2` or `:l1` and strength of the L2
     regulariser if `penalty` is `:en`."
-    lambda::Real             = 1.0
+    lambda::Real             = eps()
     "strength of the L1 regulariser if `penalty` is `:en`."
     gamma::Real              = 0.0
     "the penalty to use, either `:l2`, `:l1`, `:en` (elastic net) or `:none`."
@@ -69,7 +69,7 @@ $(example_docstring("LogisticClassifier", nclasses = 3))
 @with_kw_noshow mutable struct MultinomialClassifier <: MMI.Probabilistic
     "strength of the regulariser if `penalty` is `:l2` or `:l1`.
     Strength of the L2 regulariser if `penalty` is `:en`."
-    lambda::Real             = 1.0
+    lambda::Real             = eps()
     "strength of the L1 regulariser if `penalty` is `:en`."
     gamma::Real              = 0.0
     "the penalty to use, either `:l2`, `:l1`, `:en` (elastic net) or `:none`."