Awesome software for AutoML

A curated list of awesome AutoML frameworks.

Frameworks and libraries

🐍 Python

Machine Learning

• TPOT - Data Science Assistant - A Python Automated Machine Learning tool that optimizes machine learning pipelines using genetic programming. http://epistasislab.github.io/tpot/

• Featuretools - An open source python framework for automated feature engineering. https://www.featuretools.com

• Auto-sklearn - Automated Machine Learning with scikit-learn¹. https://automl.github.io/auto-sklearn

• Lightwood - Lightwood is Legos for Machine Learning.

• MindsDB Native - Machine Learning in one line of code. http://mindsdb.com

• mljar-supervised - Automated Machine Learning for Supervised Tasks: Binary Classification, MultiClass Classification and Regression (work in progress). https://mljar.com

Deep learning

• Auto-Keras - Accessible AutoML for deep learning. http://autokeras.com/

• NNI - An open source AutoML toolkit for neural architecture search and hyper-parameter tuning. https://nni.readthedocs.io/en/latest/

• AdaNet - Fast and flexible AutoML with learning guarantees. https://adanet.readthedocs.io

• Ludwig - Ludwig is a toolbox built on top of TensorFlow that allows to train and test deep learning models without the need to write code. http://ludwig.ai

• darts - Differentiable architecture search for convolutional and recurrent networks. https://arxiv.org/abs/1806.09055

Hybrid solutions

• automl-gs - Provide an input CSV and a target field to predict, generate a model + code to run it. https://github.com/minimaxir/automl-gs

• MLBox - MLBox is a powerful Automated Machine Learning python library. https://mlbox.readthedocs.io/en/latest/

• mindsdb - In-Database Machine Learning. MindsDB automates and abstracts machine learning models through virtual AI Tables.

R

Machine Learning

• forester - Full automation of the process of training tree-based models, wrapped into a single train() function.

Deep learning

• R interface to Auto-Keras - Package: R interface to Auto-Keras. http://autokeras.com/

Scala

Machine Learning

• TransmogrifAI - is an AutoML library for building modular, reusable, strongly typed machine learning workflows on Apache Spark with minimal hand-tuning. https://transmogrif.ai

Java

Machine Learning

• Glaucus - A general data-flow-based machine learning suit combining auto machine learning and multiple simplified machine learning algorithm for "unprofessional" data scenitists

Supplementary tools

Java

• Fione: Fess AutoML - AutoML extension for Fess.

Python

• PocketFlow - An Automatic Model Compression (AutoMC) framework for developing smaller and faster AI applications. https://pocketflow.github.io

• Ray - A fast and simple framework for building and running distributed applications. Ray is packaged with RLlib, a scalable reinforcement learning library, and Tune, a scalable hyperparameter tuning library. https://ray.readthedocs.io/en/latest/

• SMAC3 - Sequential Model-based Algorithm Configuration. https://www.automl.org/automated-algorithm-design/algorithm-configuration/smac/

¹Warning: auto-sklearn seems to be unstable at the moment - Travis build is failing and lot of unresolved issues (>100)