How to train models using Google ML ( keras & tenserflow)
You should provide more information about the previous framework you used and your layer setup. I imagine you also want to "translate" your trained model into a proper weight format