search

jrieke / traingenerator

🧙 A web app to generate template code for machine learning
https://traingenerator.jrieke.com
MIT License
1.37k stars 181 forks source link

New template: Image classification with Keras/Tensorflow #7

Open jrieke opened 3 years ago

jrieke commented 3 years ago

Opening up this issue to coordinate work on a new keras/tensorflow template – @jaymody and @EteimZ have shown interest in this. Guys, please feel free to share here if/how you want to work on this and coordinate :)

The template could probably take a lot of the code from the existing pytorch template and sklearn template for image classification and just train tensorflow models. I guess keras will be the easiest (and most useful) option.

EteimZ commented 3 years ago

Thanks @jrieke, I will open a Work in Progress PR as soon as possible.