ENH: default n_epochs for Latent models is now 100 (#122)

artefactory · Jul 10, 2024 · a1ec487 · a1ec487
1 parent be1a5b5
commit a1ec487
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/choice_learn/models/latent_class_mnl.py b/choice_learn/models/latent_class_mnl.py
@@ -1,4 +1,5 @@
 """Latent Class MNL models."""
+
 import copy
 
 import tensorflow as tf
@@ -15,7 +16,7 @@ def __init__(
         self,
         n_latent_classes,
         fit_method,
-        epochs,
+        epochs=100,
         add_exit_choice=False,
         tolerance=1e-6,
         intercept=None,
@@ -127,7 +128,7 @@ def __init__(
         n_latent_classes,
         fit_method,
         coefficients=None,
-        epochs=1,
+        epochs=100,
         add_exit_choice=False,
         tolerance=1e-6,
         optimizer="Adam",