Skip to main content
Use the Arize Python package to create datasets, run experiments, get traces, and log evaluations to Arize AX. You can also use our ML features to log predictions and actuals. See our API reference below.

Python SDK API Reference Documentation

​
Installing the package

pip install arize

Changelog

​
Available packages

The main package is arize, and you can add additional functionality by adding extra dependencies. Python 3.8 is the minimum required for the additional packages.
PackageWhat It’s For
arizeThe primary Arize package for interfacing with Arize AX APIs.
arize[Tracing]Components for logging traces and running experiments for LLM applications.
arize[AutoEmbeddings]Automatically generate embeddings vectors for your predictions and actuals for ML models.
arize[NLP_Metrics]Calculate evaluation metrics for your NLP Generative tasks.
arize[MimicExplainer]Produce SHAP values using the surrogate explainability approach.