ctf4science.data_module.get_prediction_timesteps#
- ctf4science.data_module.get_prediction_timesteps(dataset_name: str, pair_id: int, subset: str = 'test') ndarray#
Return time steps at which prediction matrices must be evaluated after training.
Uses dataset metadata (matrix shape, start index,
delta_t) to build a 1D array of time steps for the given subset matrix of the pair.- Parameters:
- dataset_namestr
Name of the dataset (e.g.
'ODE_Lorenz','PDE_KS').- pair_idint
ID of the train-test pair.
- subset{‘test’, ‘initialization’}, optional
Which matrix to use. Default is
'test'.
- Returns:
- ndarray
1D array of time steps for evaluation.
- Raises:
- ValueError
If pair_id is missing, subset matrix or metadata (e.g.
matrix_shapes,matrix_start_index,delta_t) is missing.