Gemini Enterprise Agent PlatformのCustom Training試してみた

Gemini Enterprise Agent PlatformのCustom Training機能を利用してscikit-learnモデルを学習した事例を紹介。ローカル環境での学習コード実行と比較し、マネージドVM上でのコンテナ実行の利便性を解説。

• Gemini Enterprise Agent PlatformのCustom Trainingは、ローカル環境の学習実行環境構築やログ管理を不要にする。
• 学習コードは指定コンテナをマネージドVM上で実行し、ログはCloud Loggingに自動収集される。
• 学習済みモデルなどのArtifactは自動保存されないため、Cloud Storage URIを取得し明示的に保存する必要がある。
• Cloud Buildで学習用Imageを作成し、Custom Jobでモデル学習、結果をCloud Storageに保存する構成が示されている。
• 再現性検証のため、同一条件での2回実行における評価指標、ログ、モデルファイルハッシュ値、ファイルサイズ、Cloud Storage Generationを比較している。