tensorflow-js:如何显示图形?

tensorflow-js: How to display a graph?

我调试了一些 tfjs 代码,这些代码描述了具有简单操作但混合了占位符和可训练项的图形。对我的模型进行图形可视化非常有用,例如 https://www.tensorflow.org/guide/graph_viz 提供的。

javascriptAPI里面有这样的东西吗?

tfjs-vis 有一个方法可以呈现模型拓扑的 HTML table 摘要: https://github.com/tensorflow/tfjs-vis#showmodelsummarycontainer-drawable-model-tfmodel--promise

但是TF.js目前不提供模型拓扑的图形渲染。

有一些解决方法,

  1. 您可以保存TF.js模型,使用tfjs converter and use Keras' plot_model方法将其转换为Keras模型以生成模型拓扑的png图像

  2. 使用TensorSpace(第三方工具):https://tensorspace.org/