ユースケース
アーティファクトは、ML ワークフロー全体を通じて run の入力および出力として使用できます。データセットやモデル、さらには他のアーティファクトも、処理の入力として使用できます。
以下のスニペットは、順番に実行することを前提としています。
アーティファクトを作成する
4行のコードでアーティファクトを作成します:- W&B run を作成します。
wandb.Artifactでアーティファクト オブジェクトを作成します。wandb.Artifact.add_file()で、モデルファイルやデータセットなどの1つ以上のファイルをアーティファクト オブジェクトに追加します。wandb.Run.log_artifact()でアーティファクトをW&Bにログします。
dataset.h5 というファイルを example_artifact というアーティファクトにログする方法を示しています:
- アーティファクトの
typeは、W&B プラットフォーム上での表示方法に影響します。typeを指定しない場合、デフォルトでunspecifiedになります。 - ドロップダウンの各ラベルは、それぞれ異なる
typeパラメーターの値を表しています。上記のコードスニペットでは、アーティファクトのtypeはdatasetです。
Amazon S3 バケットなどの外部オブジェクトストレージに保存されているファイルやディレクトリへの参照を追加する方法については、外部ファイルをトラッキングする ページを参照してください。
アーティファクトをダウンロードする
run にアーティファクトを入力としてマークしたい場合は、wandb.Run.use_artifact() メソッドを使用します。
前のコードスニペットに続いて、次のコード例は、以前に作成した example_artifact という名前のアーティファクトの使い方を示しています:
root パラメーター にカスタムのパスを渡すことで、アーティファクトを特定のディレクトリにダウンロードできます。アーティファクトをダウンロードする代替の方法や追加のパラメーターについては、アーティファクトのダウンロードと使用 のガイドを参照してください。