search

UMCUtrecht-ECGxAI / ecgxai

Neatly packaged AI methods for explainable ECG analysis
GNU Affero General Public License v3.0
47 stars 10 forks source link

Is beta=32 for the best run?? #2

Closed saloni-nferx closed 1 year ago

saloni-nferx commented 1 year ago

https://github.com/rutgervandeleur/ecgxai/blob/be05b53fe5f59dcea9b629f541aca3a1b981c545/examples/vae/train_factorecg_vae.py#L74

Its not very clear in the publication what's the beta being used for the optimal run,

The two most important hyperparameters in the β-VAE were the number of ECG factors and the β-value. For both, values of 8, 16, 32, 64, and 128 were evaluated.

saloni-nferx commented 1 year ago

@rutgervandeleur

rutgervandeleur commented 1 year ago

Hi Saloni! 32 was indeed the best beta in our dataset.