Question

我正在尝试使用绘图来可视化梯度下降方法。我想添加动画以查看GD如何调整最小值。问题是，我无法并排设置3D和2D绘图的动画（我希望同时可视化轮廓绘图和3d绘图并排）。我正在使用以下代码进行可视化

fig = make_subplots(rows=1, cols=2, specs=[[{"type": "scene"}, {"type": "xy"}]])



fig.add_trace(go.Scatter3d(x=trace[:, 0], y=trace[:, 1], z=trace[:, 2],
                           marker=dict(size=4, colorscale='Viridis'),
                           line=dict(color='red', width=2)),
              row=1, col=1)

fig.add_trace(go.Surface(x=x_, y=y_, z=z_, opacity=0.9, showscale=False),
              row=1, col=1)

fig.add_trace(go.Contour(z=z_, x=x_, y=y_, contours=dict(showlabels=True)),
              row=1, col=2)

fig.add_trace(go.Scatter(x=trace[:, 0], y=trace[:, 1], line=dict(color='red', width=2)),
              row=1, col=2)

fig.update_layout(width=1200, height=600, autosize=False,
                  title_text="Gradient descent demonstration",
                  #scene=dict(aspectratio = dict(x=1, y=1, z=1)),
                  showlegend=False)
fig.show()

此处“迹线”包含GD点，而“ x _”，“ y_”和“ z_”是表面点。主要问题是，我无法向其中任何一个添加属性“框架”。我已经考虑过使用小部件，但是它们太慢了。没有官方文档中的信息或任何示例。我将不胜感激。

绘制3D + 2D绘图动画

0 个答案: