Skip to content

Towards Unified Sequence Representation VII: Add MIMIC-IV YAIB example for Unified_MLP and Unified_LogisticRegression - #1356

Draft
Logiquo wants to merge 1 commit into
feat/seq-v2-6from
feat/seq-v2-7
Draft

Logiquo wants to merge 1 commit into
feat/seq-v2-6from
feat/seq-v2-7

Conversation

@Logiquo

@Logiquo Logiquo commented Oct 30, 2026 •

Copy link
Copy Markdown
Collaborator

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

examples/yaib_mimic4_unified_mlp.py trains Unified_MLP (the default) or Unified_LogisticRegression (--model logistic_regression) on each of the five YAIB MIMIC-IV tasks. It follows the same split, preflight, training, checkpoint and reporting flow as the Unified_RNN example.

@Logiquo
Logiquo added this pull request to stack #1339 October 30, 2026 11:41
@Logiquo
Logiquo marked this pull request as draft October 30, 2026 11:42
@github-actions

github-actions Bot commented Oct 30, 2026 •

Copy link
Copy Markdown

Test Coverage

Total: 75.21% (21499 / 28584 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 · 8925e7e

examples/yaib_mimic4_unified_mlp.py trains Unified_MLP (default) or, with
--model logistic_regression, Unified_LogisticRegression on each of the
five YAIB MIMIC-IV tasks 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