PyHealth
PyHealth copied to clipboard
Add BHCToAVS model for patient-friendly summaries
Contributor: Charan Williams ([email protected])
Contribution Type: New Model
Description:
Added a new clinical summarization model, BHCToAVS, which converts Brief Hospital Course (BHC) notes into patient-friendly After-Visit Summaries (AVS). The model wraps a fine-tuned Mistral-7B LoRA adapter hosted on Hugging Face and integrates with the PyHealth model API. This contribution includes the full model implementation, unit tests, documentation, and an example usage script.
Files to Review:
pyhealth/models/bhc_to_avs.py— Main model implementationpyhealth/models/__init__.py— Added import for the new modeltests/core/test_bhc_to_avs.py— Unit test for the BHCToAVS modeldocs/api/models/pyhealth.models.bhc_to_avs.rst— Sphinx documentation filedocs/api/models.rst— Updated model index to include BHCToAVSexamples/bhc_to_avs_example.py— Example usage demonstrating model prediction