fusionlab.utils.generate_dummy_pinn_data

fusionlab.utils.generate_dummy_pinn_data(n_samples, *, year_range=None, coords_range=None, subs_range=None, gwl_range=None, rainfall_range=None, vars_range=None)[source]

Generate dummy PINN data dictionary with specified or default ranges.

Parameters:

  • n_samples (int) – Number of samples to generate.

  • year_range (tuple[float, float], optional) – (min_year, max_year) for integer years. Default (2000, 2025).

  • coords_range (tuple[tuple[float, float], tuple[float, float]], optional) – ((lon_min, lon_max), (lat_min, lat_max)). Default ((113.0, 113.8), (22.3, 22.8)).

  • subs_range (tuple[float, float], optional) – (mean_subsidence, std_subsidence) for normal distribution. Default (-20, 15).

  • gwl_range (tuple[float, float], optional) – (mean_gwl, std_gwl) for normal distribution. Default (2.5, 1.0).

  • rainfall_range (tuple[float, float], optional) – (min_rain, max_rain) for uniform distribution. Default (500, 2500).

  • vars_range (dict, optional) – Dictionary that may contain any of the keys: ‘year_range’, ‘coords_range’, ‘subs_range’, ‘gwl_range’, ‘rainfall_range’. Missing keys will fall back to defaults or to explicitly passed arguments.

Returns:

dummy_data_dict –

Dictionary with keys:

  • ”year” : integer years array

  • ”longitude” : float longitudes array

  • ”latitude” : float latitudes array

  • ”subsidence” : float subsidence values array

  • ”GWL” : float groundwater level values array

  • ”rainfall_mm” : float rainfall values array

Return type:

dict[str, np.ndarray]