#161 · Primary category: Deep Learning Frameworks
ktrain
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
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.