Add in capability of train test split by brain

[nlsschim]

Create a function that, if there is enough data, trajectories in training sets and testing+validation are from unique brains (i.e., trajectories from brain X, Y, Z are in training and trajectories from brain W are in testing/validation).