pykoop.split_episodes

split_episodes(X, episode_feature=False)

Split a data matrix into episodes.

Parameters:

• X (np.ndarray) – Data matrix.

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

Returns:

List of episode tuples. The first element of each tuple contains the episode index. The second element contains the episode data.

Return type:

List[Tuple[float, np.ndarray]]