如何繪製張量流神經網絡對象

Question

我嘗試了張量流螺旋數據集的示例代碼並構建了神經網絡。我想要可視化網絡的圖形結構。 PFB我試過的代碼。

with tf.Graph().as_default(): 
    net = tflearn.input_data([None, 2]) 
    net = tflearn.fully_connected(net,6, 
       activation='tanh',weights_init='normal') 
    print(net) 

Answer 1

要使圖形可視化，您應該使用TensorBoard。以下是關於如何使用它的tutorial。

您可以在代碼的末尾添加一個彙總寫入程序，該程序會將一個事件文件（包含圖形的可視化）寫入給定位置。

graph = tf.Graph() 
with graph.as_default(): 
    net = tflearn.input_data([None, 2]) 
    net = tflearn.fully_connected(net,6, 
       activation='tanh',weights_init='normal') 

sess = tf.Session(graph=graph) 
writer = tf.train.SummaryWriter('tmp/tensorboard_log', sess.graph) 

然後你只需要運行tensorboard --logdir tmp/tensorboard_log走在瀏覽器中localhost:6006/#graphs。