Visdom stem()绘制茎叶图（附带实例）

Visdom 是 PyTorch 深度学习最常用的模型可视化工具。Visdom 提供了多种绘图函数，可以用于实现数据的可视化。

stem() 函数可绘制一个茎叶图，它接受一个 N 或 N×M 张量 X 作为输入，它指定 M 时间序列中 N 个点的值。还可以指定一个包含时间戳的可选 N 或 N×M 张量 Y，如果 Y 是一个 N 张量，那么所有 M 个时间序列都假设有相同的时间戳。

下面是该函数支持的选项：

opts.colormap：色图（string; default = 'Viridis'）；
opts.legend：包含图例名称的表；
opts.layoutopts：图形后端为布局接受的任何附加选项的字典，比如 layoutopts={'plotly':{'legend': {'x':0, 'y':0}}}。

以下是绘制一个茎叶图例子的代码：

# 导入相关库
import math
import visdom
import numpy as np
# 生成一个在0和2π之间等间距的 70 个点的数列
Y = np.linspace(0, 2 * math.pi, 70)
# 将正弦函数和余弦函数的对应值堆叠在一起
X = np.column_stack((np.sin(Y), np.cos(Y)))
# 使用Visdom库绘制茎叶图
vis.stem(
    # X 轴数据
    X=X,
    # Y 轴数据
    Y=Y,
    # 图表选项
    opts=dict(legend=['正弦函数', '余弦函数'])
)

输出如下图所示：

图 1 茎叶图

Visdom stem()绘制茎叶图（附带实例）

相关文章