/pmdarima

A package that brings R's beloved auto.arima to Python, making an even stronger case for why Python > R for data science.

Primary LanguagePythonMIT LicenseMIT

PyPI version Linux build status Build status CircleCI codecov Supported versions

pmdarima

Pmdarima (originally pyramid-arima, for the anagram of 'py' + 'arima') is a no-nonsense statistical Python library with a solitary objective: bring R's auto.arima functionality to Python. Pmdarima operates by wrapping statsmodels.tsa.ARIMA and statsmodels.tsa.statespace.SARIMAX into one estimator class and creating a more user-friendly estimator interface for programmers familiar with scikit-learn.

Installation

Pmdarima is on pypi under the package name pmdarima and can be downloaded via pip:

$ pip install pmdarima

Note that legacy versions (<1.0.0) are available under the name "pyramid-arima" and can be pip installed via:

# Legacy warning:
$ pip install pyramid-arima
# python -c 'import pyramid;'

To ensure the package was built correctly, import the following module in python:

from pmdarima.arima import auto_arima

Availability

pmdarima is available in pre-built Wheel files for the following Python versions:

  • Python 3.5+:
    • Mac (64-bit)
    • Linux (64-bit manylinux)
    • Windows (32 & 64-bit)

If a wheel doesn't exist for your platform, you can still pip install and it will build from the source distribution tarball, however you'll need cython>=0.28 and gcc (Mac/Linux) or MinGW (Windows) in order to build the package from source.

Documentation

All of your questions and more (including examples and guides) can be answered by the pmdarima documentation. If not, always feel free to file an issue.