flashML 0.1.3

AutoML tool

flashML - AutoML tool

flashML is a AutoML Python library that finds most accurate machine learning models automatically and efficiently. It frees users from selecting models and hyper-parameters for each model.

Installation

pip install flashML

Quickstart

from flashML import autoML
aml = autoML()
aml.fit(X_train, X_test, y_train, y_test, "classification", "f1_score")

Task can be either classification or regression and metric can be selected accordingly.

hyper-parameter optimization is done using optuna.

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

aml.get_best_model()

You can use predict() function for custom predicitions.

aml.predict(X_val)

Change Log

0.1.3 (16/11/2021)

• First Release