Wandb项目中如何通过Python代码设置折线图Y轴最大值

在机器学习实验跟踪工具Wandb的使用过程中，数据可视化是一个非常重要的功能。本文将详细介绍如何通过Python代码设置折线图Y轴的最大值，帮助开发者更好地控制数据展示范围。

问题背景

在Wandb的可视化面板中，用户经常需要调整折线图的Y轴范围以获得更好的数据展示效果。虽然可以通过网页界面手动调整，但在自动化实验流程中，能够通过代码预设这些参数会更加高效。

解决方案

Wandb提供了程序化工作区(programmatic workspaces)功能，允许开发者通过Python代码配置各种可视化参数。对于折线图Y轴范围的设置，可以使用range_y参数：

wr.LinePlot(range_y=(min_value, max_value))

完整示例代码

以下是一个完整的示例，展示如何创建Wandb项目、记录数据并设置折线图的Y轴范围：

import wandb
import wandb_workspaces.workspaces as ws
import wandb_workspaces.reports.v2 as wr
from datetime import datetime, timezone

# 初始化Wandb
wandb.login()

# 创建项目并记录数据
def create_project_and_log_data():
    with wandb.init(project="MyProject") as run:
        for step in range(100):
            wandb.log({
                "Step": step,
                "val_loss": 1.0 / (step + 1),
                "val_accuracy": step / 100.0,
                "train_loss": 1.0 / (step + 2),
                "train_accuracy": step / 110.0,
            })
    return "MyProject"

project = create_project_and_log_data()
entity = wandb.Api().default_entity

# 创建工作区并设置Y轴范围
def create_workspace_with_y_range(entity, project):
    workspace = ws.Workspace(name="Custom Y Range Workspace", 
                           entity=entity, 
                           project=project)
    
    workspace.sections = [
        ws.Section(
            name="Loss Section",
            panels=[
                wr.LinePlot(x="Step", y=["train_loss"], range_y=(0.0, 1.0)),
                wr.LinePlot(x="Step", y=["val_loss"], range_y=(0.0, 1.0)),
            ]
        ),
        ws.Section(
            name="Accuracy Section",
            panels=[
                wr.LinePlot(x="Step", y=["train_accuracy"], range_y=(0.0, 1.0)),
                wr.LinePlot(x="Step", y=["val_accuracy"], range_y=(0.0, 1.0)),
            ]
        ),
    ]
    
    workspace.save()

create_workspace_with_y_range(entity, project)

技术细节

1. range_y参数：该参数接受一个包含最小值和最大值的元组，用于固定Y轴的显示范围。

2. 工作区配置：通过Workspace和Section类可以构建复杂的可视化面板布局。

3. 数据类型支持：此方法适用于各种数值型数据，包括损失值、准确率等常见指标。

注意事项

1. 确保使用的Wandb库版本支持程序化工作区功能。

2. 设置的范围应合理包含数据的实际范围，否则可能导致数据点显示不全。

3. 对于多曲线图表，设置的Y轴范围应能容纳所有曲线的数据范围。

通过这种方法，开发者可以在实验脚本中预设可视化参数，实现完全自动化的实验跟踪和结果展示流程。