Skip to content

Commit

Permalink
Update test_func_ge.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Barry57 authored Oct 26, 2024
1 parent e4de949 commit 81e8ef7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pytest/test_func_ge.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,6 @@
from GENetLib.func_ge import func_ge

def test_func_ge():
func_continuous = sim_data_func(n=50, m=5, ytype='Continuous', seed=123)
func_continuous = sim_data_func(n=100, m=5, 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)
assert func_ge_res is not None

0 comments on commit 81e8ef7

Please sign in to comment.