Top AI Repos — open-source AI, indexed and scored
Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
🧙 A web app to generate template code for machine learning
| Date | Stars |
|---|---|
| 2026-07-24 | 1382 |
| 2026-07-25 | 1382 |
| 2026-07-28 | 1382 |
| 2026-07-30 | 1382 |
| 2026-07-31 | 1382 |
| 2026-08-06 | 1382 |
Today
— stars today
This week
— stars this week
This month
— stars this month
Momentum
0.0
growth rate 0.00%/day
<h1 align="center">
Traingenerator
</h1>
<p align="center">
<strong>🧙 A web app to generate template code for machine learning ✨</strong>
</p>
<p align="center">
<!-- <a href="https://twitter.com/jrieke"><img src="https://img.shields.io/twitter/follow/jrieke?color=blue&label=Follow%20%40jrieke&logo=twitter&style=plastic"></a> -->
<a href="https://traingenerator.streamlitapp.com/"><img src="https://static.streamlit.io/badges/streamlit_badge_black_white.svg"></a>
<!-- <a href="https://www.buymeacoffee.com/jrieke"><img src="https://img.shields.io/badge/Buy%20me%20a-coffee-orange.svg?logo=buy-me-a-coffee&logoColor=orange"></a> -->
<!-- <a href="LICENSE"><img src="https://img.shields.io/github/license/jrieke/traingenerator.svg"></a> -->
<a href="https://github.com/psf/black"><img src="https://img.shields.io/badge/code%20style-black-000000.svg" alt="Code style: black"></a>
</p>
<br>
<p align="center">
<img src="docs/assets/demo.gif" width=600>
</p>
<br>
<h3 align="center">
🎉 Traingenerator is now live! 🎉
<br><br>
Try it out: <br>
<a href="https://traingenerator.streamlitapp.com/">https://traingenerator.streamlitapp.com/</a>
</h3>
<!--
<p align="center"><strong>
Try it out: <br>
>>> <a href="https://traingenerator.streamlitapp.com/">https://traingenerator.streamlitapp.com/</a> <<<</strong>
</p>
-->
<br>
Generate custom template code for PyTorch & sklearn, using a simple web UI built with [streamlit](https://www.streamlit.io/). Traingenerator offers multiple options for preprocessing, model setup, training, and visualization (using Tensorboard or comet.ml). It exports to .py, Jupyter Notebook, or [Google Colab](https://colab.research.google.com/). The perfect tool to jumpstart your next machine learning project!
<br>
*For updates, [follow me on Twitter](https://twitter.com/jrieke).*
<br>
---
<br>
## Adding new templates
You can add your own template in 4 easy steps (see below), without changing any code
in the app itself. Your new template will be automatically discovered by Traingenerator
and shown in the sidebar. That's it! 🎈
**Want to share your magic?** 🧙 PRs are welcome! Please have a look
at [CONTRIBUTING.md](CONTRIBUTING.md).
**Some ideas for new templates:** Keras/Tensorflow, Pytorch Lightning, object detection,
segmentation, text classification, ...
<img align="right" src="docs/assets/dropdowns.png" width=160>
1. **Create a folder under `./templates`.**
The folder name should be the task that your template solves (e.g.
`Image classification`). Optionally, you can add a framework name (e.g.
`Image classification_PyTorch`). Both names are automatically shown in the first two
dropdowns in the sidebar (see image).
✨ *Tip: Copy the [example template](templates/example) to get started more quickly.*
1. **Add a file `sidebar.py` to the folder ([see example](templates/example/sidebar.py)).**
It needs to contain a method `show()`, which displays all template-specific streamlit
components in the sidebar (i.e. everything below *Task*) and returns a dictionary of
user inputs.
3. **Add a file `code-template.py.jinja` to the folder ([see example](templates/example/code-template.py.jinja)).**
This [Jinja2 template](https://jinja.palletsprojects.com/en/2.11.x/templates/) is used
to generate the code. You can write normal Python code in it and modify it
(through Jinja) based on the user inputs in the sidebar (e.g. insert a parameter
value from the sidebar or show different code parts based on the user's selection).
4. **Optional: Add a file `test-inputs.yml` to the folder ([see example](templates/example/test-inputs.yml)).**
This simple YAML file should define a few possible user inputs that can be used for
testing. If you run pytest (see below), it will automatically pick up this file, render
the code template with its values, and check that the generated code runs without
errors. This file is optional – but it's required if you want to Excerpt of 6,320 characters
Read on GitHubWould you bet a product on this? Bounded 0–100 and slow moving.
matched fp:c83c64146a82c9e4, topic:deep-learning, topic:pytorch, topic:tensorflow
