From 4a5e95a4a08b849df7576375552dde279b45cb76 Mon Sep 17 00:00:00 2001 From: Barry <160387047+Barry57@users.noreply.github.com> Date: Wed, 22 Jan 2025 16:11:56 +0800 Subject: [PATCH] Update test_grid_scalar_ge.py --- pytest/test_grid_scalar_ge.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytest/test_grid_scalar_ge.py b/pytest/test_grid_scalar_ge.py index be87788..ff51cc0 100644 --- 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