AutoML tool
Project description
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
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
flashML-0.1.3.tar.gz
(4.8 kB
view hashes)