#161 · Primary category: Deep Learning Frameworks

ktrain

computer-vision deep-learning graph-neural-networks keras machine-learning nlp python tabular-data tensorflow

ktrain is a Python library that makes deep learning and AI more accessible and easier to apply

Project last updated:02/06/25

GitHub Stars

1.3K

Forks

260

Contributors

17

License

Apache-2.0

Why we included this project

ktrain wraps TensorFlow Keras so you can train a neural network without building the whole stack yourself. Pick one of its pre-built learners for text, images, graphs, or tabular data, point it at your dataset, and it handles model construction, training, and prediction in a handful of lines. That makes it a good fit for teams that want a working baseline fast, whether that is BERT or fastText text classification, an image classifier, or node classification on a graph, before they invest in a hand-built pipeline. Newcomers get an easier path into Keras, while people who already know the framework can still drop down to the underlying layers when they need control. Because it produces standard Keras models, what you train stays portable and straightforward to deploy.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category