A Python package for fast exploration of machine learning pipelines

ATOM is an open-source, easy-to-use machine learning package for Python. ATOM is capable of running experiments quickly and efficiently, enabling the user to go from raw data to generating insights in just a few lines of code.

Read more Get started

Why you should use ATOM

  • Avoid endless imports and documentation lookups
  • Multiple data cleaning and feature engineering classes
  • 35+ classification and regression models to choose from
  • Possibility to train multiple models with one line of code
  • Fast implementation of hyperparameter tuning
  • Easy way to compare the results from different models
  • 35+ plots to analyze the data and model performance
  • Avoid refactoring to test new pipelines

Read our stories

tds

ATOM: A Python package for fast exploration of machine learning pipelines

During the exploration phase of a 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...

Read more
tds

How to test multiple machine learning pipelines with just a few lines of Python

Since it's nearly impossible to know beforehand which transformations will benefit the model's outcome the most, this process usually involves trying out different approaches. For example, if we are dealing with an imbalanced dataset, should we oversample the minority class or undersample the majority...

Read more
tds

From raw data to web app deployment with ATOM and Streamlit

In this article we will show you how to create a simple web app, capable of helping a data scientist to quickly perform a basic analysis on the performance of predictive models on a provided dataset. The user will be able to upload his own dataset and tweak the machine learning pipeline in two ways: selecting which data cleaning steps to apply...

Read more
Back to top