@@ -5641,8 +5641,7 @@ <h2 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor" class
56415641< span class ="normal "> 281</ span >
56425642< span class ="normal "> 282</ span >
56435643< span class ="normal "> 283</ span >
5644- < span class ="normal "> 284</ span >
5645- < span class ="normal "> 285</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> class</ span > < span class ="nc "> NNFunnelRegressor</ span > < span class ="p "> (</ span > < span class ="n "> L</ span > < span class ="o "> .</ span > < span class ="n "> LightningModule</ span > < span class ="p "> ):</ span >
5644+ < span class ="normal "> 284</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> class</ span > < span class ="nc "> NNFunnelRegressor</ span > < span class ="p "> (</ span > < span class ="n "> L</ span > < span class ="o "> .</ span > < span class ="n "> LightningModule</ span > < span class ="p "> ):</ span >
56465645< span class ="w "> </ span > < span class ="sd "> """</ span >
56475646< span class ="sd "> A LightningModule class for a regression neural network model.</ span >
56485647< span class ="sd "> This is a funnel shape neural network with varying number of layers and neurons per layer. An enhanced version of this class is available</ span >
@@ -5746,7 +5745,6 @@ <h2 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor" class
57465745 < span class ="k "> if</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> hparams</ span > < span class ="o "> .</ span > < span class ="n "> l1</ span > < span class ="o "> <</ span > < span class ="mi "> 8</ span > < span class ="p "> :</ span >
57475746 < span class ="k "> raise</ span > < span class ="ne "> ValueError</ span > < span class ="p "> (</ span > < span class ="s2 "> "l1 must be at least 8"</ span > < span class ="p "> )</ span >
57485747
5749-
57505748 < span class ="n "> layers</ span > < span class ="o "> =</ span > < span class ="p "> []</ span >
57515749 < span class ="n "> in_features</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> _L_in</ span >
57525750 < span class ="n "> hidden_size</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> hparams</ span > < span class ="o "> .</ span > < span class ="n "> l1</ span >
@@ -6268,8 +6266,7 @@ <h3 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor.__init
62686266< span class ="normal "> 131</ span >
62696267< span class ="normal "> 132</ span >
62706268< span class ="normal "> 133</ span >
6271- < span class ="normal "> 134</ span >
6272- < span class ="normal "> 135</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="fm "> __init__</ span > < span class ="p "> (</ span >
6269+ < span class ="normal "> 134</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="fm "> __init__</ span > < span class ="p "> (</ span >
62736270 < span class ="bp "> self</ span > < span class ="p "> ,</ span >
62746271 < span class ="n "> l1</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span >
62756272 < span class ="n "> num_layers</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span >
@@ -6335,7 +6332,6 @@ <h3 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor.__init
63356332 < span class ="k "> if</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> hparams</ span > < span class ="o "> .</ span > < span class ="n "> l1</ span > < span class ="o "> <</ span > < span class ="mi "> 8</ span > < span class ="p "> :</ span >
63366333 < span class ="k "> raise</ span > < span class ="ne "> ValueError</ span > < span class ="p "> (</ span > < span class ="s2 "> "l1 must be at least 8"</ span > < span class ="p "> )</ span >
63376334
6338-
63396335 < span class ="n "> layers</ span > < span class ="o "> =</ span > < span class ="p "> []</ span >
63406336 < span class ="n "> in_features</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> _L_in</ span >
63416337 < span class ="n "> hidden_size</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> hparams</ span > < span class ="o "> .</ span > < span class ="n "> l1</ span >
@@ -6411,7 +6407,8 @@ <h3 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor.config
64116407
64126408 < details class ="quote ">
64136409 < summary > Source code in < code > spotpython/light/regression/nn_funnel_regressor.py</ code > </ summary >
6414- < div class ="highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> 250</ span >
6410+ < div class ="highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> 249</ span >
6411+ < span class ="normal "> 250</ span >
64156412< span class ="normal "> 251</ span >
64166413< span class ="normal "> 252</ span >
64176414< span class ="normal "> 253</ span >
@@ -6445,8 +6442,7 @@ <h3 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor.config
64456442< span class ="normal "> 281</ span >
64466443< span class ="normal "> 282</ span >
64476444< span class ="normal "> 283</ span >
6448- < span class ="normal "> 284</ span >
6449- < span class ="normal "> 285</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> configure_optimizers</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> optim</ span > < span class ="o "> .</ span > < span class ="n "> Optimizer</ span > < span class ="p "> :</ span >
6445+ < span class ="normal "> 284</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> configure_optimizers</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> optim</ span > < span class ="o "> .</ span > < span class ="n "> Optimizer</ span > < span class ="p "> :</ span >
64506446< span class ="w "> </ span > < span class ="sd "> """</ span >
64516447< span class ="sd "> Configures the optimizer for the model.</ span >
64526448
@@ -6555,7 +6551,8 @@ <h3 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor.forwar
65556551
65566552 < details class ="quote ">
65576553 < summary > Source code in < code > spotpython/light/regression/nn_funnel_regressor.py</ code > </ summary >
6558- < div class ="highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> 137</ span >
6554+ < div class ="highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> 136</ span >
6555+ < span class ="normal "> 137</ span >
65596556< span class ="normal "> 138</ span >
65606557< span class ="normal "> 139</ span >
65616558< span class ="normal "> 140</ span >
@@ -6566,8 +6563,7 @@ <h3 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor.forwar
65666563< span class ="normal "> 145</ span >
65676564< span class ="normal "> 146</ span >
65686565< span class ="normal "> 147</ span >
6569- < span class ="normal "> 148</ span >
6570- < span class ="normal "> 149</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> forward</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> x</ span > < span class ="p "> :</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> :</ span >
6566+ < span class ="normal "> 148</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> forward</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> x</ span > < span class ="p "> :</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> :</ span >
65716567< span class ="w "> </ span > < span class ="sd "> """</ span >
65726568< span class ="sd "> Performs a forward pass through the model.</ span >
65736569
@@ -6681,7 +6677,8 @@ <h3 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor.predic
66816677
66826678 < details class ="quote ">
66836679 < summary > Source code in < code > spotpython/light/regression/nn_funnel_regressor.py</ code > </ summary >
6684- < div class ="highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> 221</ span >
6680+ < div class ="highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> 220</ span >
6681+ < span class ="normal "> 221</ span >
66856682< span class ="normal "> 222</ span >
66866683< span class ="normal "> 223</ span >
66876684< span class ="normal "> 224</ span >
@@ -6707,8 +6704,7 @@ <h3 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor.predic
67076704< span class ="normal "> 244</ span >
67086705< span class ="normal "> 245</ span >
67096706< span class ="normal "> 246</ span >
6710- < span class ="normal "> 247</ span >
6711- < span class ="normal "> 248</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> predict_step</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> batch</ span > < span class ="p "> :</ span > < span class ="nb "> tuple</ span > < span class ="p "> ,</ span > < span class ="n "> batch_idx</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span > < span class ="n "> prog_bar</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> :</ span >
6707+ < span class ="normal "> 247</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> predict_step</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> batch</ span > < span class ="p "> :</ span > < span class ="nb "> tuple</ span > < span class ="p "> ,</ span > < span class ="n "> batch_idx</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span > < span class ="n "> prog_bar</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> :</ span >
67126708< span class ="w "> </ span > < span class ="sd "> """</ span >
67136709< span class ="sd "> Performs a single prediction step.</ span >
67146710
@@ -6837,7 +6833,8 @@ <h3 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor.test_s
68376833
68386834 < details class ="quote ">
68396835 < summary > Source code in < code > spotpython/light/regression/nn_funnel_regressor.py</ code > </ summary >
6840- < div class ="highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> 204</ span >
6836+ < div class ="highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> 203</ span >
6837+ < span class ="normal "> 204</ span >
68416838< span class ="normal "> 205</ span >
68426839< span class ="normal "> 206</ span >
68436840< span class ="normal "> 207</ span >
@@ -6851,8 +6848,7 @@ <h3 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor.test_s
68516848< span class ="normal "> 215</ span >
68526849< span class ="normal "> 216</ span >
68536850< span class ="normal "> 217</ span >
6854- < span class ="normal "> 218</ span >
6855- < span class ="normal "> 219</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> test_step</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> batch</ span > < span class ="p "> :</ span > < span class ="nb "> tuple</ span > < span class ="p "> ,</ span > < span class ="n "> batch_idx</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span > < span class ="n "> prog_bar</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> :</ span >
6851+ < span class ="normal "> 218</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> test_step</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> batch</ span > < span class ="p "> :</ span > < span class ="nb "> tuple</ span > < span class ="p "> ,</ span > < span class ="n "> batch_idx</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span > < span class ="n "> prog_bar</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> :</ span >
68566852< span class ="w "> </ span > < span class ="sd "> """</ span >
68576853< span class ="sd "> Performs a single test step.</ span >
68586854
@@ -6941,7 +6937,8 @@ <h3 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor.traini
69416937
69426938 < details class ="quote ">
69436939 < summary > Source code in < code > spotpython/light/regression/nn_funnel_regressor.py</ code > </ summary >
6944- < div class ="highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> 169</ span >
6940+ < div class ="highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> 168</ span >
6941+ < span class ="normal "> 169</ span >
69456942< span class ="normal "> 170</ span >
69466943< span class ="normal "> 171</ span >
69476944< span class ="normal "> 172</ span >
@@ -6954,8 +6951,7 @@ <h3 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor.traini
69546951< span class ="normal "> 179</ span >
69556952< span class ="normal "> 180</ span >
69566953< span class ="normal "> 181</ span >
6957- < span class ="normal "> 182</ span >
6958- < span class ="normal "> 183</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> training_step</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> batch</ span > < span class ="p "> :</ span > < span class ="nb "> tuple</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> :</ span >
6954+ < span class ="normal "> 182</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> training_step</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> batch</ span > < span class ="p "> :</ span > < span class ="nb "> tuple</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> :</ span >
69596955< span class ="w "> </ span > < span class ="sd "> """</ span >
69606956< span class ="sd "> Performs a single training step.</ span >
69616957
@@ -7071,7 +7067,8 @@ <h3 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor.valida
70717067
70727068 < details class ="quote ">
70737069 < summary > Source code in < code > spotpython/light/regression/nn_funnel_regressor.py</ code > </ summary >
7074- < div class ="highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> 185</ span >
7070+ < div class ="highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> 184</ span >
7071+ < span class ="normal "> 185</ span >
70757072< span class ="normal "> 186</ span >
70767073< span class ="normal "> 187</ span >
70777074< span class ="normal "> 188</ span >
@@ -7087,8 +7084,7 @@ <h3 id="spotpython.light.regression.nn_funnel_regressor.NNFunnelRegressor.valida
70877084< span class ="normal "> 198</ span >
70887085< span class ="normal "> 199</ span >
70897086< span class ="normal "> 200</ span >
7090- < span class ="normal "> 201</ span >
7091- < span class ="normal "> 202</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> validation_step</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> batch</ span > < span class ="p "> :</ span > < span class ="nb "> tuple</ span > < span class ="p "> ,</ span > < span class ="n "> batch_idx</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span > < span class ="n "> prog_bar</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> :</ span >
7087+ < span class ="normal "> 201</ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span class ="k "> def</ span > < span class ="nf "> validation_step</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> batch</ span > < span class ="p "> :</ span > < span class ="nb "> tuple</ span > < span class ="p "> ,</ span > < span class ="n "> batch_idx</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span > < span class ="n "> prog_bar</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> Tensor</ span > < span class ="p "> :</ span >
70927088< span class ="w "> </ span > < span class ="sd "> """</ span >
70937089< span class ="sd "> Performs a single validation step.</ span >
70947090
0 commit comments