Dependencies

---

Python & OS

As of the moment, ATOM supports the following Python versions:

• Python 3.10
• Python 3.11
• Python 3.12

And operating systems:

• Linux (Ubuntu, Fedora, etc...)
• Windows 8.1+
• macOS (not tested)

Packages

Required

ATOM is built on top of several existing Python libraries. These packages are necessary for its correct functioning.

• beartype (>=0.18.5)
• category-encoders (>=2.6.3)
• dill (>=0.3.6)
• featuretools (>=1.28.0)
• gplearn (>=0.4.2)
• imbalanced-learn (>=0.12.3)
• ipython (>=8.9.0)
• ipywidgets (>=8.1.1)
• joblib (>=1.3.1)
• matplotlib (>=3.7.2)
• mlflow (>=2.10.2)
• nltk (>=3.8.1)
• numpy (>=1.23.0)
• optuna (>=3.6.0)
• pandas (>=2.2.2)
• plotly (>=5.18.0)
• scikit-learn (>=1.5.0)
• scipy (>=1.10.1)
• shap (>=0.43.0)
• sktime[forecasting] (>=0.30.0)
• statsmodels (>=0.14.1)
• zoofs (>=0.1.26)

Optional

Some specific models, utility methods or plots require the installation of additional libraries. You can install all the optional dependencies using pip install atom-ml[full]. Doing so also installs the following libraries:

• botorch (>=0.8.5)
• catboost (>=1.2)
• dagshub (>=0.3.8)
• dask[dataframe,distributed] (>=2024.2.0)
• explainerdashboard (>=0.4.3)
• gradio (>=3.44.4)
• lightgbm (>=4.1.0)
• modin[ray] (>=0.25.0)
• optuna-integration (>=3.6.0)
• polars (>=0.20.7)
• pyarrow (>=15.0.0)
• pyspark (>=3.5.0)
• requests (>=2.31.0)
• ray[serve] (>=2.9.1)
• requests (>=2.31.0)
• scikit-learn-intelex (>=2023.2.1)
• schemdraw (>=0.16)
• statsforecast (>=1.7.3)
• supersmoother (>=0.4)
• sweetviz (>=2.3.1)
• wordcloud (>=1.9.2)
• xgboost (>=2.0.0)

Development

The development dependencies are not installed with the package, and are not required for any of its functionalities. These libraries are only necessary to contribute to the project. Install them running pdm install --dev (remember to install pdm first with pip install -U pdm).

Linting

• pre-commit (>=3.6.2)

Testing

• nbmake (>=1.5.3)
• pytest (>=8.1.1)
• pytest-cov (>=4.1.0)
• pytest-mock (>=3.12.0)
• pytest-xdist (>=3.5.0)
• scikeras (>=0.13.0)
• tensorflow (>=2.16.1)

Documentation

• jupyter-contrib-nbextensions (>=0.7.0)
• kaleido (>=0.2.1)
• mike (>=2.0.0)
• mkdocs (>=1.5.3)
• mkdocs-autorefs (>=1.0.1)
• mkdocs-git-revision-date-localized-plugin (>=1.2.4)
• mkdocs-jupyter (>=0.24.6)
• mkdocs-material (>=9.5.13)
• mkdocs-material-extensions (>=1.3.1)
• mkdocs-simple-hooks (>=0.1.5)
• notebook (==6.4.12)
• pymdown-extensions (>=10.7.1)
• pyyaml (>=6.0.1)