English

ktrain: A Low-Code Library for Augmented Machine Learning

Machine Learning 2022-04-07 v5 Computation and Language Computer Vision and Pattern Recognition Social and Information Networks

Abstract

We present ktrain, a low-code Python library that makes machine learning more accessible and easier to apply. As a wrapper to TensorFlow and many other libraries (e.g., transformers, scikit-learn, stellargraph), it is designed to make sophisticated, state-of-the-art machine learning models simple to build, train, inspect, and apply by both beginners and experienced practitioners. Featuring modules that support text data (e.g., text classification, sequence tagging, open-domain question-answering), vision data (e.g., image classification), graph data (e.g., node classification, link prediction), and tabular data, ktrain presents a simple unified interface enabling one to quickly solve a wide range of tasks in as little as three or four "commands" or lines of code.

Keywords

Cite

@article{arxiv.2004.10703,
  title  = {ktrain: A Low-Code Library for Augmented Machine Learning},
  author = {Arun S. Maiya},
  journal= {arXiv preprint arXiv:2004.10703},
  year   = {2022}
}

Comments

9 pages