diff --git a/docs/pages/fitting_nn.ipynb b/docs/pages/fitting_nn.ipynb index 86656b5..cacdf63 100644 --- 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" ] },