deepof.annotation_utils.augment_with_neighbors

deepof.annotation_utils.augment_with_neighbors(X_huddle, window=5, step=1, window_out=11)

Expands a given set of features with leading and lagging features on the time axis. Will only return speed based features.

Parameters:

  • X_huddle (pandas.DataFrame) – mouse features over time.

  • window (int) – steps to go forward and backward in time for each feature

  • step (int) – step size for the window

  • window_out (int) – total length of the output window

Returns:

mouse features over time including leading and lagging features (only speed features) for each frame

Return type:

X_augmented (pandas.DataFrame)