update c2st-nn default mlp sizes

mackelab · Feb 6, 2024 · d98fb56 · d98fb56
1 parent 0aa57fe
commit d98fb56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/labproject/metrics/c2st.py b/labproject/metrics/c2st.py
@@ -75,7 +75,7 @@ def c2st_nn(
     ndim = X.shape[-1]
     defaults = {
         "activation": activation,
-        "hidden_layer_sizes": (10 * ndim, 10 * ndim),
+        "hidden_layer_sizes": (min(10 * ndim, 100), min(10 * ndim, 100)),
         "max_iter": 1000,
         "solver": "adam",
         "early_stopping": True,