Python SDK
Arize AI for Model Monitoring, Troubleshooting, and Explainability
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.
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:
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]
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]
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]
The Arize Python SDK offers 2 ways of logging data into the platform:
Designed for logging a batch of your model inferences using Pandas DataFrames. Go to the following page for more information.
Designed for low latency, one-at-a-time, logging of your model inferences. Go to the following page for more information.
Major Release | First Released | Latest | Support |
|---|---|---|---|
7.x | June, 2023 | 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 |