Dependencies
Python & OS
As of the moment, ATOM supports the following Python versions:
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.
- category-encoders (>=2.4.1)
- explainerdashboard (>=0.4.0)
- dill (>=0.3.5)
- evalml (>=0.62.0)
- gplearn (>=0.4.1)
- gradio (>=3.3.1)
- imbalanced-learn (>=0.9.0)
- featuretools (>=1.14.0)
- joblib (>=1.1.0, <1.2.0)
- matplotlib (>=3.5.0, <3.6.0)
- mlflow (>=2.0.1)
- nltk (>=3.7)
- numpy (>=1.22)
- optuna (>=3.0.0)
- pandas (>=1.3.5)
- pandas-profiling (>=3.5.0)
- plotly (>=5.10.0)
- shap (>=0.41)
- schemdraw (>=0.14)
- scikit-learn (>=1.1.0)
- scikit-learn-intelex (>=2021.6.3)
- scipy (>=1.8.1)
- typeguard (>=2.13)
- wordcloud (>=1.8.1)
- zoofs (>=0.1.24)
Optional
You can install some optional packages to be able to use some well-known
machine learning estimators that are not provided by sklearn but are
among ATOM's predefined models. Install them using pip install atom-ml[models].
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
using pip install atom-ml[dev].
Linting
- isort (>=5.10.1)
- flake8 (>=5.0.4)
- flake8-pyproject (>=1.1.0)
Testing
- pytest (>=7.1.0)
- pytest-cov (>=3.0.0)
Documentation
- mike (>=1.1.2)
- mkdocs (>=1.2.3)
- mkdocs-autorefs (>=0.4.1)
- mkdocs-jupyter (>=0.22.0)
- mkdocs-material (>=8.5.3)
- mkdocs-simple-hooks (>=0.1.5)
- pyyaml (>=6.0)