機械学習の 実験 でこれらのクラスを使用する詳細については、 Experiments および Artifacts のドキュメントを参照してください。
実験の追跡( Experiments )やアーティファクト( Artifacts )の管理を行うために、W&B Python SDK の基本クラスを使用します。
機械学習の 実験 でこれらのクラスを使用する詳細については、 Experiments および Artifacts のドキュメントを参照してください。
import wandb
# 新しい run を初期化します
with wandb.init(project="my-experiments", config={"learning_rate": 0.001}) as run:
# 設定にアクセスします
config = run.config
# トレーニング中のメトリクスをログに記録します
for epoch in range(10):
metrics = train_one_epoch() # トレーニングロジック
run.log({
"loss": metrics["loss"],
"accuracy": metrics["accuracy"],
"epoch": epoch
})
# サマリーメトリクスを記録します
run.summary["best_accuracy"] = max_accuracy
import wandb
with wandb.init(project="my-models") as run:
# モデルをトレーニングします
model = train_model()
# モデルの Artifact を作成します
model_artifact = wandb.Artifact(
name="my-model",
type="model",
description="ResNet-50 trained on ImageNet subset",
metadata={
"architecture": "ResNet-50",
"dataset": "ImageNet-1K",
"accuracy": 0.95
}
)
# Artifact にモデルファイルを追加します
model_artifact.add_file("model.pt")
model_artifact.add_dir("model_configs/")
# Artifact を W&B にログ記録します
run.log_artifact(model_artifact)
import wandb
# プログラムから設定を構成します
wandb.Settings(
project="production-runs",
entity="my-team",
mode="offline", # オフラインで実行し、後で同期します
save_code=True, # ソースコードを保存します
quiet=True # コンソール出力を抑制します
)
# または環境変数を使用します
# export WANDB_PROJECT=production-runs
# export WANDB_MODE=offline
# カスタム設定で初期化します
with wandb.init() as run:
# ここに実験のコードを記述します
pass
import wandb
with wandb.init(project="ml-pipeline") as run:
# データセット Artifact を使用します
dataset = run.use_artifact("dataset:v1")
dataset_dir = dataset.download()
# データセットを使用してモデルをトレーニングします
model = train_on_dataset(dataset_dir)
# データセットのリネージを持つモデル Artifact を作成します
model_artifact = wandb.Artifact(
name="trained-model",
type="model"
)
model_artifact.add_file("model.pt")
# 自動的なリネージトラッキングを伴うログ記録を行います
run.log_artifact(model_artifact)