pykoop.extract_initial_conditions

extract_initial_conditions(X, min_samples=1, n_inputs=0, episode_feature=False)

Extract initial conditions from each episode.

Parameters:

• X (np.ndarray) – Data matrix.

• min_samples (int) – Number of samples in initial condition.

• n_inputs (int) – Number of input features at the end of X.

• episode_feature (bool) – True if first feature indicates which episode a timestep is from.

Returns:

Initial conditions from each episode.

Return type:

np.ndarray