TensorBoard或其他工具可以可视化TensorFlow.js模型吗？

Question

在https://www.tensorflow.org/js/guide/save_load中，描述了将模型文件另存为使用model.json和相应的model.weights.bin的格式。我不确定是否有谈论这种格式的名称（我认为它与https://js.tensorflow.org/api/latest/#class:LayersModel相同，但并不完全确定），我想知道是否有一种方法可以将它们可视化为图表。

我原本希望能够在TensorBoard中加载和查看它们，但是看不到使用其“图形”工具执行此操作的任何方法，所以也许还没有人做过这样的事情。

Answer 1

执行此操作的一种最小方法是在已加载的模型上使用summary方法，该方法将记录到控制台：

在我尝试的示例中，此输出与查看model.json时的预期输出不匹配。看起来这可能只查看最外层的顺序层，但我没有仔细看。

Answer 2

您可以通过Visor API https://js.tensorflow.org/api_vis/latest/检查模型，层和张量等。您只需要安装npm i @tensorflow/tfjs-vis即可使用。