ADD: updates notebook

artefactory · Nov 28, 2024 · f5b4171 · f5b4171
1 parent 40db13b
commit f5b4171
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/notebooks/models/latent_class_model.ipynb b/notebooks/models/latent_class_model.ipynb
@@ -79,7 +79,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "lc_model = LatentClassSimpleMNL(n_latent_classes=3, fit_method=\"mle\", optimizer=\"lbfgs\", epochs=1000, tolerance=1e-20)\n",
+    "lc_model = LatentClassSimpleMNL(n_latent_classes=3, fit_method=\"mle\", optimizer=\"lbfgs\", epochs=1000, lbfgs_tolerance=1e-20)\n",
     "hist, results = lc_model.fit(elec_dataset, verbose=1)"
    ]
   },
@@ -134,7 +134,7 @@
     "                                       fit_method=\"mle\",\n",
     "                                       optimizer=\"lbfgs\",\n",
     "                                       epochs=1000,\n",
-    "                                       tolerance=1e-12)"
+    "                                       lbfgs_tolerance=1e-12)"
    ]
   },
   {
@@ -280,7 +280,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": "tf_env",
    "language": "python",
    "name": "python3"
   },
@@ -294,7 +294,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.18"
+   "version": "3.11.4"
   }
  },
  "nbformat": 4,