PyHealth
PyHealth copied to clipboard
TCN Bounty Implementation
Bilal Arif netID: barif
Contribution Type: Updated model
Description:
- Updated to current BaseModel/EmbeddingModel API
- Implemented dilated causal convolutions for temporal sequence modeling on EHR data
- Supports both discrete code sequences (diagnoses, procedures) and continuous timeseries
- Includes comprehensive test suite with 6 passing unit tests
- Added example notebook demonstrating mortality prediction on MIMIC-III
Files to review:
- pyhealth/models/tcn.py - Full TCN implementation
- tests/core/test_tcn.py - Main testing suite
- examples/tcn_mimic3_codes.ipynb - Jupyter notebook for using MIMIC-III