python - 如何绘制不同训练样本的置信区间

Question

我正在与训练集的不同部门一起进行训练。我得到的图（使用 wandb）很好，但在我看来信息量不是很大，而且方差很大。

有没有办法绘制图的平均值，然后围绕它的置信区间？类似于下图的东西。或者，有没有办法在训练期间绘制方差？

Answer 1

group如果您使用W&B 的功能，您可以获得您想要的情节。当您初始化 W&B 运行时，将一些值传递给group参数。

run = wandb.init(entity='your-id', 
                 project='your-project-name', 
                 config='config-dict', 
                 group='group-name' # example: training-set-name
                 job_type='train' # what this W&B run is logging
)

在此处查看文档。

示例图：

您还可以编辑对实验进行分组的不同方式。

附带说明：您也可以使用不同的平滑方法。文档在这里。