Skip to content

HeartDisease dataset & task#726

Closed
runalih wants to merge 1 commit intosunlabuiuc:masterfrom
runalih:master
Closed

HeartDisease dataset & task#726
runalih wants to merge 1 commit intosunlabuiuc:masterfrom
runalih:master

Conversation

@runalih
Copy link
Copy Markdown

@runalih runalih commented Dec 8, 2025

In this PR we are including 3 new files that aim to add a new dataset and a task to go along with that dataset.

This dataset is Heart_Disease_UCI.csv. It contains many columns that all pertains to a patients current health status (e.g. age, sex, cholesterol, blood pressure) and their current heart problems. Using this csv file, we’ve created a Dataset that is imported into pyhealth to interpret this data.

This data is then pulled into a task that aims to use the columns in this dataset to try to predict where or not patients have any heart issues. This task also provides a train/testing split in order to test if the model is accurate.

New Files added/modified:
pyhealth/datasets/heart_dataset.py
pyhealth/datasets/rawdata/heart_disease_uci.csv
pyhealth/tasks/heart_task.py

Examples of usages:
# pull in datasets from local

dataset = HeartDiseaseDataset(root="pyhealth/datasets/rawdata/heart_disease_uci.csv")

# run tasks on dataset to predict information

task = HeartDiseasePrediction(dataset)



# output information on splits

(train_X, train_y), (test_X, test_y) = task.split()

…ks for predicting heart problems from dataset.
@Logiquo Logiquo added component: dataset Contribute a new dataset to PyHealth component: task Contribute a new task to PyHealth labels Dec 18, 2025
@Logiquo Logiquo changed the title Init Commit, adding files for importing heart disease dataset and tas… HeartDisease dataset & task Dec 18, 2025
@Logiquo
Copy link
Copy Markdown
Collaborator

Logiquo commented Dec 27, 2025

Could you add the test case for the new dataset and tasks? thank you.

@Logiquo Logiquo added the status: wait response Pending PR author's response label Dec 27, 2025
@github-actions
Copy link
Copy Markdown

This PR has been automatically marked as stale because it has not had recent activity. It will be closed in 7 days if no further activity occurs.

@github-actions github-actions Bot added the stale label Mar 28, 2026
@github-actions
Copy link
Copy Markdown

github-actions Bot commented Apr 4, 2026

This PR has been automatically closed due to inactivity. Please re-open it if you think it is still relevant.

@github-actions github-actions Bot closed this Apr 4, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

component: dataset Contribute a new dataset to PyHealth component: task Contribute a new task to PyHealth stale status: wait response Pending PR author's response

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants