u/affnn

Essentially, I am looking at a data set with many individuals. Each individual has a number of observation dates associated with them (sometimes one date, sometimes multiple dates), and each observation date is a "data point". I want to separate the train/test data by the individuals, but keep the observation dates as "data points".

Here's a hypothetical version:

I have 30 individuals and 50 readings. Each individual has 1, 2 or 3 readings. I want to do a train/test split so I have 20 individuals in the train and 10 in the test. I don't care as much how many readings I have in each set, but the default method would be to partition the 50 readings without regard to the individuals.

Is there a way I can do that with this package?