jrieke / traingenerator

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

Adding tensorflow_template #9

Open EteimZ opened 3 years ago

EteimZ commented 3 years ago

Summary

This PR adds template for Image Classification in Tensorflow.

Details

### Checklist - [x] all tests are passing (see README.md on how to run tests) - [x] if you created a new template: it contains a file `test-inputs.yml`, which specifies a few input values to test the code template (the test is then automatically run by pytest) - [x] you formatted all code with [black](https://github.com/psf/black) - [x] you checked all new functionality live, i.e. in the running web app - [x] any generated code is formatted nicely, both in .py and in .ipynb ("nicely" = comparable to the existing templates) - [x] you added comments in your code that explain what it does - [x] the PR explains in detail what's new