0.16.7

regression max length 10

Author: bartzbeielstein

pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "spotpython"
-version = "0.16.6"
+version = "0.16.7"
 authors = [
   { name="T. Bartz-Beielstein", email="tbb@bartzundbartz.de" }
 ]

src/spotpython/light/regression/nn_linear_regressor.py

@@ -249,10 +249,13 @@ def _generate_div2_list(self, n, n_min) -> list:
             repeats = repeats + 1
         return result
 
-    def _get_hidden_sizes(self):
+    def _get_hidden_sizes(self, n=10) -> list:
         n_low = self._L_in // 4
         n_high = max(self.hparams.l1, 2 * n_low)
         hidden_sizes = self._generate_div2_list(n_high, n_low)
+        # keep only the first 10 values of hidden_sizes list
+        if len(hidden_sizes) > n:
+            hidden_sizes = hidden_sizes[:n]
         return hidden_sizes
 
     def forward(self, x: torch.Tensor) -> torch.Tensor: