search

flatironinstitute / catvae

Categorical Variational Autoencoders
BSD 3-Clause "New" or "Revised" License
22 stars 3 forks source link

Adding per-dimension over-dispersion terms #66

Closed mortonjt closed 3 years ago

mortonjt commented 3 years ago

Before, we were having the model MLN(WW'). This will add an additional diagonal term MLN(WW' + K)

mortonjt commented 3 years ago

This works so we'll merge this in.