FlashML-AutoML tool

flashML is a AutoML Python library that finds the most accurate machine learning models automatically and efficiently. It frees data scientists and ML engineers from worrying about model selection and hyperparameter tuning.

Automated machine learning (AutoML) is the process of applying machine learning (ML) models to real-world problems using automation. More specifically, it automates the selection, composition and parameterization of machine learning models.

Installation

Quickstart

The task can be either classification or regression and metric can be selected accordingly.

optuna is used for hyperparameter optimization.

After training, use this function to get the best model:

You can use predict function for custom predictions, it uses the best model to make predictions.

GitHub link: https://github.com/omchaithanyav/flashML