[CS598] New dataset loader class for IAFib

[huiyinz]

Author

Huiyin Zheng - huiyinz2 ([email protected])

Usage

from pyhealth.datasets.iafib import iAFibDataset
# Initialize the dataset
root = "~/pyhealth-test" # Make it the location where you want the data to be saved
config_path = "<path to your PyHealth Repo> /PyHealth/pyhealth/datasets/configs/iAFib.yaml" # Location of the yaml file
dataset = iAFibDataset(
    root=root,
    config_path=config_path
)
print(dataset.global_event_df.head()) # View first 5 rows of the data

[dalloliogm]

Hi, in a recent PR, I've added a notebook with a list of datasets used in PyHealth. Should I add this one as well? https://github.com/sunlabuiuc/PyHealth/pull/616