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.

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:

LLM Tracing

minimum required for LLM Tracing

With this extra module, Arize can receive your traces and spans to break down your system's components into discrete inputs and outputs to help debug your LLM application. Learn more here.

pip install arize[Tracing]

Auto Embeddings

minimum required for Auto Embeddings

With this extra module, Arize properly extracts the embeddings 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]

NLP Metrics

minimum required for NLP Metrics

With this extra module, Arize helps you calculate evaluation metrics for your NLP Generative tasks. Learn more here. To install the Arize package including this functionality:

pip install arize[NLP_Metrics]

Mimic Explainer

minimum required for Mimic Explainer

With this extra module, Arize allows the user 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: