pykoop.extract_input

extract_input(X, n_inputs=0, episode_feature=False)

Extract input from a data matrix.

Parameters:

• X (np.ndarray) – Data matrix.

• 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:

Input extracted from data matrix.

Return type:

np.ndarray