Update test_grid_scalar_ge.py

Barry57 · Oct 28, 2024 · 08d6685 · 08d6685
1 parent e7b5e22
commit 08d6685
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytest/test_grid_scalar_ge.py b/pytest/test_grid_scalar_ge.py
@@ -12,8 +12,8 @@ def test_grid_scalar_ge():
                                         Learning_Rate2=[0.035,0.045], L2=[0.1], Learning_Rate1=[0.04, 0.05, 0.06], L=[0.08, 0.09], 
                                         Num_Epochs=50, t = 0.01, model = None, split_type = 0, ratio = [7, 3], 
                                         important_feature = True, plot = True, model_reg = None, isfunc = False)
-  grid_scalar_ge_res_2 = grid_scalar_ge(data=scalar_binary['data'], ytype='Binary', dim_G=500, dim_E=5, haveGE=False, 
-                                        num_hidden_layers=2, nodes_hidden_layer=[1000,100], 
+  grid_scalar_ge_res_2 = grid_scalar_ge(data=scalar_binary['data'], ytype='Binary', dim_G=1500, dim_E=5, haveGE=False, 
+                                        num_hidden_layers=1, nodes_hidden_layer=[50], 
                                         Learning_Rate2=[0.035,0.045], L2=[0.1], Learning_Rate1=[0.04, 0.05, 0.06], L=[0.08, 0.09], 
                                         Num_Epochs=50, t = 0.01, model = None, split_type = 1, ratio = [7, 2, 1], 
                                         important_feature = False, plot = False, model_reg = None, isfunc = False)