You can deploy this model and start earning money today!

Katja-Jagd/
bat-pretrained-eicu-mimiciii-base

apache-2.0

Views

No views yet

BAT Pretrained Model (eICU + MIMIC-III)

This repository contains the PyTorch Lightning checkpoint for the BAT_pretrained model trained on the eICU and MIMIC-III datasets. All experiments were conducted using PyTorch 2.4.0+cu118 with Python 3.10.16

Model Files

model.ckpt: PyTorch Lightning checkpoint with the trained model weights.

Usage

1. Create a new virtual environment and activate it

bash

python -m venv bat-env
source bat-env/bin/activate      # On Linux or macOS
pip install -r https://raw.githubusercontent.com/Katja-Jagd/YAIB/development/requirements.txt

2. Clone the GitHub repository

bash

git clone https://github.com/Katja-Jagd/YAIB.git
cd YAIB

3. Load the model checkpoint

bash

from icu_benchmarks.models.dl_models.bat import SSL_BAT

model = SSL_BAT.load_from_checkpoint(
    "https://huggingface.co/Katja-Jagd/bat-pretrained-eicu-mimiciii-base/resolve/main/model.ckpt"
)
model.eval()