get_discrete_chunks

behavenet.plotting.arhmm_utils.get_discrete_chunks(states, include_edges=True)

Find occurences of each discrete state.

Parameters

• states (list) – list of trials; each trial is numpy array containing discrete state for each frame

• include_edges (bool) – include states at start and end of chunk

Returns

list of length discrete states; each list contains all occurences of that discrete state by [chunk number, starting index, ending index]

Return type

list