fix res bug

arjunsavel · Mar 11, 2024 · 322ba58 · 322ba58
1 parent 2b70e2d
commit 322ba58
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/docs/pages/fitting_nn.ipynb b/docs/pages/fitting_nn.ipynb
@@ -204,6 +204,21 @@
    },
    "outputs": [],
    "source": [
+    "res = cortecs.fit.fit_neural_net.fit_neural_net(\n",
+    "    fitter.opac.cross_section[:, :, -2],\n",
+    "    fitter.opac.T,\n",
+    "    fitter.opac.P,\n",
+    "    None,\n",
+    "    n_layers=3,\n",
+    "    n_neurons=8,\n",
+    "    activation=\"sigmoid\",\n",
+    "    learn_rate=0.04,\n",
+    "    loss=\"mean_squared_error\",\n",
+    "    epochs=4000,\n",
+    "    verbose=0,\n",
+    "    sequential_model=None,\n",
+    "    plot=True,\n",
+    ")\n",
     "history, neural_network = res"
    ]
   },