wandb_workspaces から利用できる W&B Reports and Workspaces API を使用すると、調査結果の共有に向けて Web 上で公開できる Reports を作成できるほか、トレーニングやファインチューニングの作業を行った Workspace をカスタマイズできます。
W&B Report and Workspace API はパブリックプレビューです。
お使いのマシンをW&Bに対して認証するには、まず User Settings でAPIキーを発行する必要があります。
W&B Reports および Workspaces のライブラリをインストールします。
pip install wandb-workspaces
レポートを作成するには、Team の entity を指定し、レポートの名を入力します。囲まれたテキストを実際の値に置き換えてください。
import wandb_workspaces.reports.v2 as wr
# 作成
report = wr.Report(
entity="<team_entity>",
project="<project_name>",
title='Quickstart Report',
description="That was easy!"
)
# レポートを保存
report.save()
次に、レポートにブロックとパネルを追加します。たとえば、次のコードは、目次、ヘッダー、段落を含むレポートを作成します。
report.blocks = [
wr.TableOfContents(),
wr.H1("Text and images example"),
wr.P("Lorem ipsum dolor sit amet."),
]
report.save()
エンドツーエンドの例は、レポート API クイックスタート の Google Colab を参照してください。
次のコードは、折れ線グラフ、棒グラフ、スカラーチャートの 3 つのパネルを含むセクションを持つ Workspace を作成する方法を示しています。角括弧で囲まれたテキストは、実際の値に置き換えてください。
# インポート方法
import wandb_workspaces.workspaces as ws
# Workspaceを作成する
ws.Workspace(
entity="<team_entity>", # Workspaceを所有するentity
project="<project_name>", # Workspaceに関連付けられたプロジェクト
sections=[
ws.Section(
name="<Validation Metrics>",
panels=[
wr.LinePlot(x="Step", y=["<val_loss>"]),
wr.BarPlot(metrics=["<val_accuracy>"]),
wr.ScalarChart(metric="<f1_score>", groupby_aggfunc="<mean>"),
],
is_open=True,
),
],
)
workspace.save()
一連の流れを通した例については、Workspace API クイックスタート の Google Colab を参照してください。