メインコンテンツへスキップ
あらゆる規模の機械学習実験を追跡・可視化・管理するために W&B をインストールしてください。
W&B Weave についての情報をお探しですか?Weave Python SDK クイックスタート または Weave TypeScript SDK クイックスタート をご覧ください。

サインアップしてAPIキーを作成

マシンをW&Bに対して認証するには、APIキーが必要です。 APIキーを作成するには、Personal API key または Service Account API key タブを選択して詳細を確認します。
ユーザーIDに紐づく個人用APIキーを作成するには、次の手順を実行します。
  1. W&B にログインし、ユーザープロフィールアイコンをクリックしてから User Settings をクリックします。
  2. Create new API key をクリックします。
  3. APIキーのわかりやすい名前を入力します。
  4. Create をクリックします。
  5. 表示されたAPIキーをすぐにコピーし、安全な場所に保管します。
完全なAPIキーは作成時に一度だけ表示されます。ダイアログを閉じると、完全なAPIキーを再度表示することはできません。設定で表示されるのは、キーID(キーの先頭部分)のみです。完全なAPIキーを紛失した場合は、新しいAPIキーを作成する必要があります。
安全に保管する方法については、APIキーを安全に保管するを参照してください。 このクイックスタートは、Colabノートブックとしても利用できます。

wandb ライブラリをインストールしてログインする

  1. WANDB_API_KEY 環境変数 を設定します。
    export WANDB_API_KEY=<your_api_key>
    
  2. wandb ライブラリをインストールして、ログインします。
    pip install wandb
    wandb login
    

run を初期化してハイパーパラメータを追跡する

Python スクリプトまたはノートブック内で、wandb.init() を使って W&B の run オブジェクトを初期化します。config パラメータには辞書を使用して、 ハイパーパラメータ名とその値を指定します。with 文のブロック内では、メトリクスやその他の情報を W&B に記録できます。
import wandb

wandb.login()

# runが記録されるプロジェクト
project = "my-awesome-project"

# ハイパーパラメータの辞書
config = {
    'epochs' : 10,
    'lr' : 0.01
}

with wandb.init(project=project, config=config) as run:
    # 学習コードをここに記述
    # run.log() でW&Bに値を記録する
    run.log({"accuracy": 0.9, "loss": 0.1})
次のセクションでは、学習 run をシミュレートし、精度と損失のメトリクスを W&B にログする完全な例を紹介します。
run は W&B の中核要素です。run を使ってメトリクスをトラッキングし、ログを作成し、アーティファクトを追跡するなど、さまざまなことができます。

機械学習モデルの学習実験を作成する

このモック学習スクリプトでは、シミュレートされた精度 (accuracy) と損失 (loss) のメトリクスを W&B にログします。次のコードを Python スクリプトまたはノートブックセルにコピーして貼り付け、実行してください。
import wandb
import random

wandb.login()

# runの記録先プロジェクト
project = "my-awesome-project"

# ハイパーパラメータの辞書
config = {
    'epochs' : 10,
    'lr' : 0.01
}

with wandb.init(project=project, config=config) as run:
    offset = random.random() / 5
    print(f"lr: {config['lr']}")
    
    # 学習runをシミュレートする
    for epoch in range(2, config['epochs']):
        acc = 1 - 2**-config['epochs'] - random.random() / config['epochs'] - offset
        loss = 2**-config['epochs'] + random.random() / config['epochs'] + offset
        print(f"epoch={config['epochs']}, accuracy={acc}, loss={loss}")
        run.log({"accuracy": acc, "loss": loss})
wandb.ai/home にアクセスして、accuracy や loss などの記録されたメトリクスと、それらが各学習ステップでどのように変化したかを確認します。次の画像は、各 run で追跡された loss と accuracy を示しています。各 run オブジェクトは、生成された名前とともに Runs 列に表示されます。
各 run で追跡された loss と accuracy を表示しています。

次のステップ

W&B エコシステムのその他の機能も試してみてください:
  1. PyTorch のようなフレームワーク、Hugging Face のようなライブラリ、SageMaker のようなサービスと W&B を組み合わせた W&B Integration チュートリアル を読んでください。
  2. W&B Reports を使って run を整理し、可視化を自動化し、知見を要約し、共同作業者と最新情報を共有します。
  3. 機械学習パイプライン全体にわたってデータセット、モデル、依存関係、結果を追跡するために W&B Artifacts を作成します。
  4. W&B Sweeps を使ってハイパーパラメータ探索を自動化し、モデルを最適化します。
  5. 共通ダッシュボード 上で run を分析し、モデルの予測を可視化し、インサイトを共有します。
  6. W&B AI Academy にアクセスして、ハンズオンのコースを通じて LLM、MLOps、および W&B Models について学びます。
  7. weave-docs.wandb.ai にアクセスして、Weave を使って LLM ベースのアプリケーションを追跡、実験、評価、デプロイ、および改善する方法を学びます。