Skip to content

Releases: ZhiGroup/pytorch_ehr

Version 0.2

21 Feb 20:45
18d0556
Compare
Choose a tag to compare

This release is offering a faster and more memory efficient code than the previously released version

Key Changes:

  • Moving paddings and mini-batches related tensors creation to the EHR_dataloader
  • Creating the mini-batches list once before running the epochs
  • Adding RETAIN to the models list