首页 > 解决方案 > tensorflow-js:如何显示图表?

问题描述

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

javascript API中有这样的东西吗?

标签: tensorflowtensorflow.js

解决方案


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

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

有几种解决方法,

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

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


推荐阅读