dvc_studio_client_post_live_metrics_experiment_tracking.py

1from dvc_studio_client.post_live_metrics import post_live_metrics
2
3# To use this, DVC_STUDIO_TOKEN and DVC_STUDIO_REPO_URL 
4# must be set in your environment variables.
5
6post_live_metrics(
7    event="start",
8)
9
10# Your training logic here
11metrics = {"accuracy": 0.95, "loss": 0.05}
12params = {"learning_rate": 0.01, "epochs": 10}
13
14post_live_metrics(
15    event="data",
16    step=1,
17    metrics=metrics,
18    params=params,
19)
20
21post_live_metrics(
22    event="done",
23)