[Feature Request] train_test_split dataset based on replicate groups

[chirranjeevigopal-TRI]

The BeepDataset class has a method for generating train-test splits based on cell_IDs or unique sequence numbers. However, if there are replicate measurements for the same protocol, I would like to ensure that all replicates fall in the same group (train or test)

As a solution, include a method that first performs unique-parameter groupings based no the protocol parameter file, and then randomizes these groupings into train/test at prescribed level. Ensure the fraction of train vs test is maintained (even if some parameter groups have significantly more replicates than others).