PROSAGA码农传奇-模型可视化-keras模型结构可视化

keras模型结构可视化

作者: v-star*위위
发布时间: 2025-03-07 05:20:43 (5天前)
转自：

            我想看看keras

模型
</跨度>
像这样。我使用了K.get_session（）。graph和get

tensorflow.python.framework.ops.Graph位于0x7f2a8b809400

但我要看这个图并保存。我正在使用tensorflow后端

安装张量板

导入它


    
      from keras.callbacks import TensorBoard
</code>


  
    将其加载到变量中
  

   
    
      tbCallBack = TensorBoard(log_dir=’Graph’,
                         histogram_freq=10,
                         write_graph=True,
                         write_images=True)
</code>
  

  
    然后将其用作训练中的回调：
  

   
    
      model.fit(x, y, …
          callbacks=[tbCallBack])
</code>
  

  
    确保你有一个名为’Graph’的目录或任何你想要的目录。然后在终端训练之前运行：
  

   
    
      tensorboard —logdir Graph
</code>
  

  
    然后，您可以在浏览器中看到您的图表