Commit d79bb994 authored by Zhang, Chen's avatar Zhang, Chen
Browse files

expand training param range

parent 49448cca

src/tgreft/utils/data/data_loader.py

+1 −1
Original line number Diff line number Diff line
@@ -87,7 +87,7 @@ def generate_data( 
            np.random.uniform(3.5, 6.5, n_dataset),  # electolyte_sld,

            np.random.uniform(5, 120, n_dataset),  # electolyte_roughness,



            np.random.uniform(0, 6.5, n_dataset),  # sei_sld,

            np.random.uniform(-5.0, 6.5, n_dataset),  # sei_sld,

            np.random.uniform(10, 500, n_dataset),  # sei_thickness,

            np.random.uniform(1, 80, n_dataset),  # sei_roughness,