cpmpercussion / keras-mdn-layer

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

Move to tf2 api, thanks u7484835 #47

Closed cpmpercussion closed 4 months ago

cpmpercussion commented 4 months ago

This moves the code and examples to use the TensorFlow 2 API properly including eager execution in some of the RNN example notebooks.

A new test builds an MDRNN model and tests inputs and outputs by running a random prediction.

Many thanks to @u7484835 for contributing and playing with this code!