fusionlab.nn.utils.prepare_pinn_data_sequences

fusionlab.nn.utils.prepare_pinn_data_sequences(df, time_col, subsidence_col, gwl_col, dynamic_cols, static_cols=None, future_cols=None, spatial_cols=None, h_field_col=None, lon_col=None, lat_col=None, group_id_cols=None, time_steps=12, forecast_horizon=3, output_subsidence_dim=1, output_gwl_dim=1, datetime_format=None, normalize_coords=True, cols_to_scale=None, lock_physics_cols=True, protect_si_suffix='__si', return_coord_scaler=False, coord_scaler=None, fit_coord_scaler=True, mode=None, model=None, savefile=None, progress_hook=None, stop_check=None, verbose=0, _logger=None, **kws)[source]
Parameters:

  • df (DataFrame)

  • time_col (str)

  • subsidence_col (str)

  • gwl_col (str)

  • dynamic_cols (List[str])

  • static_cols (List[str] | None)

  • future_cols (List[str] | None)

  • spatial_cols (Tuple[str, str] | None)

  • h_field_col (str | None)

  • lon_col (str | None)

  • lat_col (str | None)

  • group_id_cols (List[str] | None)

  • time_steps (int)

  • forecast_horizon (int)

  • output_subsidence_dim (int)

  • output_gwl_dim (int)

  • datetime_format (str | None)

  • normalize_coords (bool)

  • cols_to_scale (List[str] | str | None)

  • lock_physics_cols (bool)

  • protect_si_suffix (str)

  • return_coord_scaler (bool)

  • coord_scaler (MinMaxScaler | None)

  • fit_coord_scaler (bool)

  • mode (str | None)

  • model (str | None)

  • savefile (str | None)

  • progress_hook (Callable[[float], None] | None)

  • stop_check (Callable[[], bool])

  • verbose (int)

  • _logger (Logger | Callable[[str], None] | None)

Return type:

Tuple[Dict[str, ndarray], Dict[str, ndarray]] | Tuple[Dict[str, ndarray], Dict[str, ndarray], MinMaxScaler | None]