Hugging Face's logo

Laiyf
/
DiffuSETS

Model card Files
xet
 Community
DiffuSETS / README.md
Laiyf's picture
Laiyf
Update README.md
249a3e6 verified
preview code
|
raw
history blame
1.16 kB
metadata 
license: mit

Model Card for DiffuSETS

Description

  • Developed by: Yongfan Lai
  • License: mit

Model Sources

Assets List

  • mimic_iv_text_embed.csv: Pre saved embedding table for mimic text report.
  • mimic_vae.pt & mimic_vae_lite.pt: Dictionary consists of VAE encoding latents of mimic iv ecg dataset. lite version contains only 50 000 samples for convenience.
  • unet.pth: A trained noise predictor model checkpoints. Use both VAE latent and patient characteristc info.
  • vae_model.pth: A trained VAE model checkpoints suit for the unet model and latent datasets above.
  • sample_config.json: An example of configuration.

Quick Uses

Download the all the prerequisites into the ./prerequisites folder, then start ECG generating through

python DiffuSETS_inference.py ./prerequisites/sample_config.json

For more details in training and inference, please refer to our github repo page