Skip to main content

State-of-the art Automated Machine Learning python library for Tabular Data

Project description

AutoML Alex

Downloads PyPI - Python Version PyPI CodeFactor Telegram License


State-of-the art Automated Machine Learning python library for Tabular Data

Works with Tasks:

  • <input type="checkbox" checked="" disabled="" />

    Binary Classification

  • <input type="checkbox" checked="" disabled="" />

    Regression

  • <input type="checkbox" disabled="" />

    Multiclass Classification (in progress...)

Benchmark Results

bench

The bigger, the better
From AutoML-Benchmark

Scheme

scheme

Features

  • Automated Data Clean (Auto Clean)
  • Automated Feature Engineering (Auto FE)
  • Smart Hyperparameter Optimization (HPO)
  • Feature Generation
  • Feature Selection
  • Models Selection
  • Cross Validation
  • Optimization Timelimit and EarlyStoping
  • Save and Load (Predict new data)

Installation

pip install automl-alex

Docs

DocPage

🚀 Examples

Classifier:

from automl_alex import AutoMLClassifier

model = AutoMLClassifier()
model.fit(X_train, y_train, timeout=600)
predicts = model.predict(X_test)

Regression:

from automl_alex import AutoMLRegressor

model = AutoMLRegressor()
model.fit(X_train, y_train, timeout=600)
predicts = model.predict(X_test)

DataPrepare:

from automl_alex import DataPrepare

de = DataPrepare()
X_train = de.fit_transform(X_train)
X_test = de.transform(X_test)

Simple Models Wrapper:

from automl_alex import LightGBMClassifier

model = LightGBMClassifier()
model.fit(X_train, y_train)
predicts = model.predict_proba(X_test)

model.opt(X_train, y_train,
    timeout=600, # optimization time in seconds,
    )
predicts = model.predict_proba(X_test)

More examples in the folder ./examples:

What's inside

It integrates many popular frameworks:

  • scikit-learn
  • XGBoost
  • LightGBM
  • CatBoost
  • Optuna
  • ...

Works with Features

  • <input type="checkbox" checked="" disabled="" />

    Categorical Features

  • <input type="checkbox" checked="" disabled="" />

    Numerical Features

  • <input type="checkbox" checked="" disabled="" />

    Binary Features

  • <input type="checkbox" disabled="" />

    Text

  • <input type="checkbox" disabled="" />

    Datetime

  • <input type="checkbox" disabled="" />

    Timeseries

  • <input type="checkbox" disabled="" />

    Image

Note

  • With a large dataset, a lot of memory is required! Library creates many new features. If you have a large dataset with a large number of features (more than 100), you may need a lot of memory.

Realtime Dashboard

Works with optuna-dashboard

Dashboard

Run

$ optuna-dashboard sqlite:///db.sqlite3

Road Map

  • <input type="checkbox" checked="" disabled="" />

    Feature Generation

  • <input type="checkbox" checked="" disabled="" />

    Save/Load and Predict on New Samples

  • <input type="checkbox" checked="" disabled="" />

    Advanced Logging

  • <input type="checkbox" checked="" disabled="" />

    Add opt Pruners

  • <input type="checkbox" checked="" disabled="" />

    Docs Site

  • <input type="checkbox" disabled="" />

    DL Encoders

  • <input type="checkbox" disabled="" />

    Add More libs (NNs)

  • <input type="checkbox" disabled="" />

    Multiclass Classification

  • <input type="checkbox" disabled="" />

    Build pipelines

Contact

Telegram Group

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for automl-alex, version 1.6.10
Filename, size File type Python version Upload date Hashes
Filename, size automl_alex-1.6.10-py3-none-any.whl (52.4 kB) File type Wheel Python version py3 Upload date Hashes View
Filename, size automl_alex-1.6.10.tar.gz (34.2 kB) File type Source Python version None Upload date Hashes View

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring DigiCert DigiCert EV certificate Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page