23-07-09

dev container 조정

mnist.py

@@ -50,7 +50,7 @@ def make_model():
 
 # %%
 model = make_model()
-x_train, y_train, x_test, y_test = get_data()
+x_train, y_train = get_data_test()
 
 loss = [
     "mse",
@@ -73,9 +73,9 @@ if __name__ == "__main__":
             loss=loss[2],
             n_particles=100,
             c0=0.35,
-            c1=0.8,
-            w_min=0.4,
-            w_max=1.1,
+            c1=0.7,
+            w_min=0.5,
+            w_max=0.9,
             negative_swarm=0.2,
             mutation_swarm=0.1,
         )