使用 Vertex AI Experiments 手動記錄指標

您可以使用 Vertex AI Experiments 執行作業,手動記錄 TensorBoard 時間序列指標。這些指標會以視覺化方式呈現於 Vertex AI Experiments 控制台,或 Vertex AI TensorBoard 實驗網頁應用程式。

如要進一步瞭解如何記錄指標和參數,請參閱「手動將資料記錄到實驗執行作業」。

Python

from typing import Dict, Optional

from google.cloud import aiplatform
from google.protobuf import timestamp_pb2


def log_time_series_metrics_sample(
    experiment_name: str,
    run_name: str,
    metrics: Dict[str, float],
    step: Optional[int],
    wall_time: Optional[timestamp_pb2.Timestamp],
    project: str,
    location: str,
):
    aiplatform.init(experiment=experiment_name, project=project, location=location)

    aiplatform.start_run(run=run_name, resume=True)

    aiplatform.log_time_series_metrics(metrics=metrics, step=step, wall_time=wall_time)

  • experiment_name:為實驗命名。
  • run_name:提供執行名稱。
  • metrics:字典,其中鍵為指標名稱,值為指標值。
  • step:選用。這個資料點在執行中的步驟索引。
  • wall_time:選用。這個資料點由使用者產生時的實際時間戳記。如未提供,系統會根據 time.time() 的值產生 wall_time
  • project:。您可以在 Google Cloud 控制台 歡迎頁面中找到這些 ID。
  • location:實驗和 TensorBoard 執行個體的位置。 如果實驗或 TensorBoard 尚不存在,系統會在該位置建立。