search

mmcdermott / EventStreamGPT

Dataset and modelling infrastructure for modelling "event streams": sequences of continuous time, multivariate events with complex internal dependencies.
https://eventstreamml.readthedocs.io/en/latest/
MIT License
102 stars 16 forks source link

PytorchDataset doesn't have a way to extract subject_ids, start_time, and end_time all together #75

Open pargaw opened 1 year ago

pargaw commented 1 year ago

Currently can extract for subject_ids, columns, and cached_data separately to get this data, though a saved dataframe/path to dataframe from the class would be helpful when referring back to specific entries.