Currently; taking in datasets of 143-146 neurons from the (WormWideWeb)[https://www.wormwideweb.org/dataset.html] datasets and training RNNs to predict behaviors
velocity: velocity trace over time
head_curvature: head curvature trace over time
pumping: pumping/feeding trace over time
angular_velocity: angular velocity trace over time
body_curvature: body curvature trace over time
reversal_events: list of reversal start/stop time points, in confocal frames