Skip to content

Towards Unified Sequence Representation V: Add MIMIC-IV YAIB example for Unified_RNN - #1354

Draft
Logiquo wants to merge 1 commit into
feat/seq-v2-4from
feat/seq-v2-5
Draft

Logiquo wants to merge 1 commit into
feat/seq-v2-4from
feat/seq-v2-5

Conversation

@Logiquo

@Logiquo Logiquo commented Oct 30, 2026 •

Copy link
Copy Markdown
Collaborator

Fifth PR in the stack toward a unified sequence representation, on top of #1353. Example only; no library changes.

examples/yaib_mimic4_unified_rnn.py trains a separate Unified_RNN on each of the five YAIB MIMIC-IV tasks (mortality, kidney function, length of stay, AKI, sepsis). It writes checkpoints, logs, the configuration and test metrics. --check verifies the data files and the device without preprocessing or training.

This is a training example using PyHealth's YAIB task definitions, not a reproduction of the original benchmark's tuning and cross-validation protocol.

@github-actions

github-actions Bot commented Oct 30, 2026 •

Copy link
Copy Markdown

Test Coverage

Total: 75.13% (21398 / 28483 lines)

Modified file Coverage Lines
pyhealth/benchmarks/__init__.py 100.00% 2 / 2
pyhealth/benchmarks/_serialization.py 95.95% 71 / 74
pyhealth/benchmarks/base.py 98.33% 59 / 60
pyhealth/models/vae.py 96.77% 90 / 93
pyhealth/tasks/__init__.py 89.74% 35 / 39
pyhealth/tasks/covid19_cxr_generation.py 100.00% 9 / 9

Workflow run · e749b3e

examples/yaib_mimic4_unified_rnn.py trains a separate Unified_RNN on each
of the five YAIB MIMIC-IV tasks (mortality, kidney function, length of
stay, AKI and sepsis) and writes checkpoints, logs, the configuration and
test metrics. --check verifies the data files and the device without
preprocessing or training.

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant