From 440a4638a9e6d55e6400c98ed87a7cc4017ded51 Mon Sep 17 00:00:00 2001 From: Barry57 <160387047+Barry57@users.noreply.github.com> Date: Sun, 27 Oct 2024 00:24:07 +0800 Subject: [PATCH] Update test_func_ge.py --- pytest/test_func_ge.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pytest/test_func_ge.py b/pytest/test_func_ge.py index ac9b0a8..e488f21 100644 --- a/pytest/test_func_ge.py +++ b/pytest/test_func_ge.py @@ -2,5 +2,8 @@ from GENetLib.func_ge import func_ge func_continuous = sim_data_func(n=100, 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=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=True) +func_ge_res = func_ge(func_continuous['y'], func_continuous['z'], func_continuous['location'], + func_continuous['X'], 'Continuous', 'Bspline', num_hidden_layers=2, nodes_hidden_layer=[20,5], + Learning_Rate2=0.035, L2=0.01, Learning_Rate1=0.02, L=0.01, Num_Epochs=1, + nbasis1=5, params1=4, Bsplines=5, norder1=4, model=None, split_type=1, ratio=[3, 1, 1], plot_res=False)