Eval Table を作成する

W&B Python SDK を使用して Eval Table を作成するか、既存の W&B Table を ARIA で変換します。

Eval Table を作成する

Eval Table を作成するには、wandb.EvalTable クラスを使用します。たとえば、次のような pandas DataFrame があるとします。

import pandas as pd
import wandb

df = pd.DataFrame(
    [
        {
            "image_id": "img_001",
            "true_label": "cat",
            "predicted_label": "cat",
            "confidence": 0.97,
            "correct": True,
        },
        {
            "image_id": "img_002",
            "true_label": "dog",
            "predicted_label": "cat",
            "confidence": 0.72,
            "correct": False,
        },
        {
            "image_id": "img_003",
            "true_label": "car",
            "predicted_label": "car",
            "confidence": 0.89,
            "correct": True,
        },
    ]
)

DataFrame を wandb.EvalTable に渡して Eval Table を作成します。input_columns、output_columns、score_columns の各引数で、各列の役割を指定します。次の例では、以下のように指定しています。

image_id を入力として使用
predicted_label を出力として使用
correct と confidence をスコアとして使用

with wandb.init(project="classifier-eval-table-demo") as run:
    eval_table = wandb.EvalTable(
        dataframe=df,
        input_columns=["image_id"],
        output_columns=["predicted_label"],
        score_columns=["correct", "confidence"],
    )
    run.log({"validation_predictions_eval": eval_table})    

W&B Table を Eval Table に変換する

変換バナーを使用して、既存の W&B Table を Eval Table に変換します。

W&B Table が含まれるプロジェクトに移動します。
変換バナーの Try now ボタンを選択します。変換対象のテーブルに関する情報とともに ARIA が開きます。
ARIA の会話ウィンドウに表示される指示に従います。

バナーを閉じてしまった場合や表示されない場合は、ARIA を手動で開きます。

ページ右上の青い丸を選択して ARIA を開きます。
ARIA の会話で /convert-eval-table と入力します。
会話の右下にある上向き矢印()の送信ボタンを選択します。

W&B Table を変換しても、元のテーブルは変更されません。

​Eval Table を作成する

​W&B Table を Eval Table に変換する

Eval Table を作成する

W&B Table を Eval Table に変換する