Search
K
Links

Python SDK

Arize AI for Model Monitoring, Troubleshooting, and Explainability
PyPI - Version
PyPI - Status
PyPI - Python Version
Use the Arize Python package to monitor machine learning predictions to observe your ML models and their features, predicted labels, and actual labels with just a few lines of code.

Installing the package

pip install arize
In addition to the basic functionality installed by the command above, the Arize SDK has additional functionality that can be installed with some extra dependencies:

Auto Embeddings

python>=3.8
minimum required for Auto Embeddings
With this extra module, Arize extracts the embeddings in the appropriate way depending on your use case, and we return it to you to include in your Pandas DataFrame. Learn more here. To install the Arize package including this functionality:
pip install arize[AutoEmbeddings]

LLM Evaluation

python>=3.8
minimum required for LLM Evaluation
With this extra module, Arize helps you calculate evaluation metrics for your LLM Generative tasks. Learn more here. To install the Arize package including this functionality:
pip install arize[LLM_Evaluation]

Mimic Explainer

With this extra module, Arize gives the user the option to pass a flag with their request to send data that would produce SHAP values using the surrogate explainability approach. Learn more here. To install the Arize package including this functionality:
pip install arize[MimicExplainer]

Logging Options

The Arize Python SDK offers 2 ways of logging data into the platform:

Pandas Batch Logging

Designed for logging a batch of your model inferences using Pandas DataFrames. Go to the following page for more information.

Single Record Logging

Designed for low latency, one-at-a-time, logging of your model inferences. Go to the following page for more information.

End of Support Table

Major Release
First Released
Latest
Support
7.x
June, 2023
latest
Ends January 1st, 2026
6.x
January, 2023
6.1.3
Ends January 1st, 2025
5.x
August, 2022
5.5.0
Ends October 1st, 2024
4.x
March, 2022
4.2.2
Ends June 1st, 2024
3.x
September, 2021
3.4.0
Ends April 1st, 2024
2.x
March, 2021
2.2.1
Ended July 1st, 2023
1.x
July, 2020
1.2.1
Ended March 1st, 2022
0.x
March, 2020
0.0.20
Ended March 1st, 2022