Create Keras code from a visual graph

This project enables you to draw a graph representing a Keras neural network, and then generate the corresponding Python code. This is a pre-release, as it is an open source project we made this site to present it anyway.

Built with the tech you love

  • Keras

    A simple Python library to build neural networks.

  • Vue.js

    Front-end framework for building user interfaces and single-page applications.

  • D3.js

    D3.js is used to draw the graph representing the Keras model.

Open Source

Released under MIT License

This project is available on GitHub. We started this as a university project, but as we're interested in it, we still work on it on our free time (although not as much as before, since we all got jobs now). If you'd like to help us, you can make pull requests for any missing functionality you could implement, and we will consider including it. We also have to refactor big parts of our D3 codebase, as we didn't have enough time to conceive it as clean as we would have liked in the first place. We could also switch our code generation output to TensorFlow.js, as it would enable the user to launch training directly in the browser, which would be simpler for educational purposes.

Demo

As it is a pre-release, not all Keras layer parameters are implemented, there are some missing functionalities, and you can experience some glitch. Still, you can try our demo.