[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

[jhnwu3]

Hey, I'm guessing you're from the class. Would like to see more things if you're not too bothered by it.

• it would be good to add docs on where the source of this dataset is with some additional details in the doc strings like how to use this dataset.

• I think if you could add a reference in the docs/ .rst files that would be really nice

• Also something like a tasks example would be cool.

No pressure though, I get just needing to fulfill credit requirements:

Here's an exemplar by Eric:
#392

[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? #616