Hi, thanks for the code. Since I dont familiar with Tensorflow directly, perhaps you have Keras version of this work?