Update test_grid_scalar_ge.py

XMU-Kuangnan-Fang-Team · Jan 22, 2025 · 4a5e95a · 4a5e95a
1 parent bde1e15
commit 4a5e95a
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
@@ -17,13 +17,13 @@ def test_grid_scalar_ge():
                                         num_hidden_layers=1, nodes_hidden_layer=[50], num_epochs=50,
                                         learning_rate1=[0.04, 0.05, 0.06], learning_rate2=[0.035,0.045],
                                         lambda1 = None, lambda2 = [0.08, 0.09], Lambda = [0.1],
-                                        threshold = 0.01, model = None, split_type = 0, ratio = [7, 2, 1], 
+                                        threshold = 0.01, model = None, split_type = 1, ratio = [7, 2, 1], 
                                         important_feature = True, plot = False, model_reg = None, isfunc = False)
   grid_scalar_ge_res_3 = grid_scalar_ge(scalar_continuous['y'], scalar_continuous['G'], scalar_continuous['E'], ytype='Survival',
                                         num_hidden_layers=2, nodes_hidden_layer=[1000,100], num_epochs=50,
                                         learning_rate1=[0.04, 0.05, 0.06], learning_rate2=[0.035,0.045],
                                         lambda1 = None, lambda2 = [0.08, 0.09], Lambda = [0.1],
-                                        threshold = 0.01, model = None, split_type = 0, ratio = [7, 2, 1], 
+                                        threshold = 0.01, model = None, split_type = 1, ratio = [7, 2, 1], 
                                         important_feature = False, plot = False, model_reg = None, isfunc = False)
   assert grid_scalar_ge_res_1 is not None
   assert grid_scalar_ge_res_2 is not None