Skip to content

Commit

Permalink
Update test_grid_func_ge.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Barry57 authored Oct 26, 2024
1 parent 6b71d5d commit 42f87e6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pytest/test_grid_func_ge.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
from GENetLib.sim_data_func import sim_data_func
from GENetLib.grid_func_ge import grid_func_ge

def test_grid_func_ge(func_continuous_data):
def test_grid_func_ge():
func_continuous = sim_data_func(n=1500, m=30, ytype='Continuous', seed=123)
grid_func_ge_res = grid_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.008, 0.009, 0.01], L2=[0.002, 0.003, 0.004, 0.005, 0.006], Learning_Rate1=[0.02, 0.03, 0.04, 0.05], L=[0.05, 0.06, 0.07, 0.08], Num_Epochs=100, nbasis1=7, params1=4, Bsplines=15, norder1=4, model=None, split_type=0, ratio=[7,3], plot_res=True)
assert grid_func_ge_res is not None

0 comments on commit 42f87e6

Please sign in to comment.