Increase arc-gen fitting data: 10→50 (fixed), 20→50 (variable) for better conv generalization

own-solver/neurogolf_solver/data_loader.py

@@ -57,7 +57,7 @@ def get_exs(td):
 
 
 def get_exs_for_fitting(td):
-    """Get examples for fitting with ARC-GEN augmentation."""
+    """Get examples for fitting with ARC-GEN augmentation (fixed shape)."""
     base_exs = [(np.array(ex['input'], dtype=np.int64), np.array(ex['output'], dtype=np.int64))
                 for ex in td['train'] + td['test']]
     if not base_exs:
@@ -72,7 +72,7 @@ def get_exs_for_fitting(td):
         out = np.array(ex['output'], dtype=np.int64)
         if inp.shape == base_shape and out.shape == base_exs[0][1].shape:
             ag_exs.append((inp, out))
-    return base_exs + ag_exs[:10]
+    return base_exs + ag_exs[:50]
 
 
 def get_exs_for_fitting_variable(td):
@@ -85,7 +85,7 @@ def get_exs_for_fitting_variable(td):
         out = np.array(ex['output'], dtype=np.int64)
         if inp.shape == out.shape and inp.shape[0] <= 30 and inp.shape[1] <= 30:
             ag_exs.append((inp, out))
-    return base_exs + ag_exs[:20]
+    return base_exs + ag_exs[:50]
 
 
 def fixed_shapes(td):