cpmpercussion / keras-mdn-layer

An MDN Layer for Keras using TensorFlow's distributions module
MIT License
165 stars 44 forks source link

Bug when multiple outputs, std missing #29

Open CoteDave opened 4 years ago

CoteDave commented 4 years ago

Hi,

When using multiple y outputs with multiple mixtures, there seem to miss some std parameters:

For example, when 2 y outputs with 4 mixtures, when calling mdn_model.predict() that gives:

I was expeting 8 stds as well, is there a reason why or it is a bug ?

Another example, if 2 y outputs with 2 mixtures: image

Thanks!