From b0805cd1d659c1fe711231b232ee5226bb1ae710 Mon Sep 17 00:00:00 2001 From: Barry57 <160387047+Barry57@users.noreply.github.com> Date: Sun, 27 Oct 2024 20:15:48 +0800 Subject: [PATCH] Update test_func_ge.py --- pytest/test_func_ge.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytest/test_func_ge.py b/pytest/test_func_ge.py index e38efcd..f2ded40 100644 --- a/pytest/test_func_ge.py +++ b/pytest/test_func_ge.py @@ -15,7 +15,7 @@ def test_func_ge(): 'Binary', 'Fourier', 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=0, ratio=[7, 3], plot_res=True) - func_ge_res_3 = func_ge(list(func_survival['y']), func_survival['z'], func_survival['location'], func_survival['X'], + func_ge_res_3 = func_ge(func_survival['y'], func_survival['z'], func_survival['location'], func_survival['X'], 'Survival', 'Power', 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=[-1,-0.5,0,0.5,1], Bsplines=5, norder1=4, model=None, split_type=0, ratio=[7, 3], plot_res=True, plot_beta = False)