Update test_func_ge.py

XMU-Kuangnan-Fang-Team · Oct 27, 2024 · 306162d · 306162d
1 parent 75778f4
commit 306162d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytest/test_func_ge.py b/pytest/test_func_ge.py
@@ -4,9 +4,9 @@
 matplotlib.use('Agg')
 
 def test_func_ge():
-  func_continuous = sim_data_func(n=50, m=30, ytype='Continuous', seed=123)
+  func_continuous = sim_data_func(n=500, m=30, ytype='Continuous', seed=123)
   func_ge_res = func_ge(func_continuous['y'], func_continuous['z'], func_continuous['location'], func_continuous['X'], 
-                        'Continuous', 'Bspline', num_hidden_layers=1, nodes_hidden_layer=[2], Learning_Rate2=0.035, 
+                        'Continuous', 'Bspline', num_hidden_layers=2, nodes_hidden_layer=[100,10], Learning_Rate2=0.035, 
                         L2=0.01, Learning_Rate1=0.02, L=0.01, Num_Epochs=50, nbasis1=5, params1=4, Bsplines=5, norder1=4, 
                         model=None, split_type=1, ratio=[3, 1, 1], plot_res=False)
   assert func_ge_res is not None