Download experiment

Experiment results can be downloaded as a Dataframe object using the Arize Python client.

To download an experiment's results, call arize_client.get_experiment on the experiment name:

# example usage
exp_df_1 = arize_client.get_experiment(space_id, experiment_name=None, dataset_name=None, experiment_id=None)

Reference:

  • space_id (str) – The ID of the space where the experiment is located.

  • experiment_name (Optional[str]) – The name of the experiment. Required if experiment_id is not provided.

  • dataset_name (Optional[str]) – The name of the dataset associated with the experiment. Required if experiment_id is not provided.

  • experiment_id (Optional[str]) – The ID of the experiment. Required if experiment_name and dataset_name are not provided.

Code Example:

# install required packages
pip install arize opentelemetry-exporter-otlp openinference-semantic-conventions
# setup

DEVELOPER_KEY = 
API_KEY = 
SPACE_ID = 
EXPERIMENT_NAME = 
DATASET_NAME = 

from arize.experimental.datasets import ArizeDatasetsClient
client = ArizeDatasetsClient(
    developer_key=DEVELOPER_KEY,
    api_key=API_KEY,
)

df = client.get_experiment(
    space_id=SPACE_ID,
    experiment_name=EXPERIMENT_NAME,
    dataset_name=DATASET_NAME,
)
# optional - save as .csv (in Google Colab)
import pandas as pd
from google.colab import files

# Save the DataFrame to CSV
df.to_csv(f"{EXPERIMENT_NAME}.csv", index=False)

# Download CSV
files.download(f"{EXPERIMENT_NAME}.csv")

Last updated

Was this helpful?