Python SDK

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.

Installing the package

pip install arize

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.

Package
What It's For

arize

The primary Arize package for interfacing with Arize AX APIs.

Components for logging traces and running experiments for LLM applications.

Automatically generate embeddings vectors for your predictions and actuals for ML models.

Calculate evaluation metrics for your NLP Generative tasks.

Produce SHAP values using the surrogate explainability approach.

Last updated

Was this helpful?