📜 Overview

Mavs      [email protected]      Documentation      Slack

General Information
Repository
Release
Compatibility
Build status
Code analysis

💡 Introduction

During the exploration phase of a machine learning project, a data scientist tries to find the optimal pipeline for his specific use case. This usually involves applying standard data cleaning steps, creating or selecting useful features, trying out different models, etc. Testing multiple pipelines requires many lines of code, and writing it all in the same notebook often makes it long and cluttered. On the other hand, using multiple notebooks makes it harder to compare the results and to keep an overview. On top of that, refactoring the code for every test can be quite time-consuming. How many times have you conducted the same action to pre-process a raw dataset? How many times have you copy-and-pasted code from an old repository to re-use it in a new use case?

ATOM is here to help solve these common issues. The package acts as a wrapper of the whole machine learning pipeline, helping the data scientist to rapidly find a good model for his problem. Avoid endless imports and documentation lookups. Avoid rewriting the same code over and over again. With just a few lines of code, it's now possible to perform basic data cleaning steps, select relevant features and compare the performance of multiple models on a given dataset, providing quick insights on which pipeline performs best for the task at hand.

Example steps taken by ATOM's pipeline:

1. Data Cleaning
   • Handle missing values
   • Encode categorical features
   • Detect and remove outliers
   • Balance the training set
2. Feature engineering
   • Create new non-linear features
   • Remove multi-collinear features
   • Remove features with too low variance
   • Select the most promising features
3. Train and validate multiple models
   • Select hyperparameters using a Bayesian Optimization approach
   • Train and test the models on the provided data
   • Assess the robustness of the output using a bootstrap algorithm
4. Analyze the results
   • Get the model scores on various metrics
   • Make plots to compare the model performances

🛠️ Installation

Install ATOM's newest release easily via pip:

$ pip install -U atom-ml

or via conda:

$ conda install -c conda-forge atom-ml

⚡ Usage

ATOM contains a variety of classes and functions to perform data cleaning, feature engineering, model training, plotting and much more. The easiest way to use everything ATOM has to offer is through one of the main classes:

• ATOMClassifier for binary or multiclass classification tasks.
• ATOMRegressor for regression tasks.

Let's walk you through an example. Click on the Google Colab badge on top of this section to run this example yourself.

Make the necessary imports and load the data.

import pandas as pd
from atom import ATOMClassifier

# Load the Australian Weather dataset
X = pd.read_csv("https://raw.githubusercontent.com/tvdboom/ATOM/master/examples/datasets/weatherAUS.csv")
X.head()

Initialize the ATOMClassifier or ATOMRegressor class. These two classes are convenient wrappers for the whole machine learning pipeline. Contrary to sklearn's API, they are initialized providing the data you want to manipulate.

atom = ATOMClassifier(X, y="RainTomorrow", test_size=0.3, verbose=2)

Data transformations are applied through atom's methods. For example, calling the impute method will initialize an Imputer instance, fit it on the training set and transform the whole dataset. The transformations are applied immediately after calling the method (no fit and transform commands necessary).

atom.impute(strat_num="median", strat_cat="most_frequent")  
atom.encode(strategy="LeaveOneOut", max_onehot=8)

Similarly, models are trained and evaluated using the run method. Here, we fit both a Random Forest and AdaBoost model, and apply hyperparameter tuning.

atom.run(models=["RF", "AdaB"], metric="auc", n_calls=10, n_initial_points=4)

Lastly, visualize the result using the integrated plots.

atom.plot_roc()
atom.rf.plot_confusion_matrix(normalize=True)

Documentation