[PYTHON] Zeigen Sie das Diagramm von tensorBoard auf jupyter an

Python / Tensorflow-Anfänger erstellen eine Jupyter + Tensorflow-Umgebung und machen Hello World - Qiita

Diesmal ist es die Prozedur, das Diagramm von tensorBoard auf jupyter anzuzeigen. Das TensorBoard ist sehr praktisch, also lassen Sie es uns so wie es ist auf dem Jupiter anzeigen! Es ist eine Prozedur.

Zielbildschirm

Erstellungsverfahren

Zur Anzeige des Tensorboard-Diagramms sind verschiedene Vorbereitungen erforderlich. Da erwartet wird, dass es viele Male verwendet wird, werden wir den störenden Teil mit dem Namen "tensorboard.py" entfernen, damit er beim Aufruf schnell verwendet werden kann.

Erstellen Sie tensorboard.py

Zuerst machen wir das Teil, um die Tensorplatine anzuzeigen

Erstellen Sie eine leere Datei

Um eine leere Datei zu erstellen, wählen Sie wie folgt Textdatei aus Neu aus (Wenn Sie dies über die Befehlszeile tun, können Sie einfach "tensorboard.py" erstellen.)

Es kann überall erstellt werden, es wird jedoch davon ausgegangen, dass es im selben Ordner wie das Notizbuch erstellt wurde.

Umbenennen und kopieren

Da es sich beim Erstellen um untitled.txt handelt, drücken Sie den Namensteil, um es zu "tensorboard.py" zu machen.

Bitte fügen Sie Folgendes für den Quellcode ein. Sie können die Größe usw. ändern.

from IPython.display import clear_output, Image, display, HTML
import tensorflow as tf
import numpy as np

def strip_consts(graph_def, max_const_size=32):
    """Strip large constant values from graph_def."""
    strip_def = tf.GraphDef()
    for n0 in graph_def.node:
        n = strip_def.node.add() 
        n.MergeFrom(n0)
        if n.op == 'Const':
            tensor = n.attr['value'].tensor
            size = len(tensor.tensor_content)
            if size > max_const_size:
                tensor.tensor_content = bytes("<stripped %d bytes>"%size, 'utf-8')
    return strip_def

def show_graph(graph_def, max_const_size=32):
    """Visualize TensorFlow graph."""
    if hasattr(graph_def, 'as_graph_def'):
        graph_def = graph_def.as_graph_def()
    strip_def = strip_consts(graph_def, max_const_size=max_const_size)
    code = """
        <script>
          function load() {{
            document.getElementById("{id}").pbtxt = {data};
          }}
        </script>
        <link rel="import" href="https://tensorboard.appspot.com/tf-graph-basic.build.html" onload=load()>
        <div style="height:600px">
          <tf-graph-basic id="{id}"></tf-graph-basic>
        </div>
    """.format(data=repr(str(strip_def)), id='graph'+str(np.random.rand()))

    iframe = """
        <iframe seamless style="width:1200px;height:620px;border:0" srcdoc="{}"></iframe>
    """.format(code.replace('"', '&quot;'))
    display(HTML(iframe))

Referenz: Einfache Möglichkeit, ein TensorFlow-Diagramm in Jupyter zu visualisieren? - Stapelüberlauf

Bestätigung

OK, wenn die Datei wie folgt erstellt wurde

Ändern Sie den Anrufer

Fügen Sie import tensorboard as tb hinzu, um die erstellte Datei tensorboard.py und tb.show_graph, den anzuzeigenden Code, zu dem Programm zu lesen, das Sie anzeigen möchten.

import tensorflow as tf
import tensorboard as tb   # ->hinzufügen

const1 = tf.constant(2)
const2 = tf.constant(3)
add_op = tf.add(const1, const2)
mul_op = tf.mul(add_op, const2)

with tf.Session() as sess:
    result, result2 = sess.run([mul_op, add_op])
    print(result)
    print(result2)

    tf.summary.FileWriter('./log/', sess.graph)
    
tb.show_graph(tf.get_default_graph().as_graph_def()) # ->hinzufügen

Danach wird es angezeigt, wenn Sie es mit Strg + Eingabetaste ausführen

Bemerkungen

Wenn Sie den Quellcode so ändern, dass er gut kopiert und eingefügt werden kann, ist die Verwendung einfacher. Und ich möchte wirklich eher Ereignisse als Diagramme anzeigen, aber ich konnte keinen Weg finden (´ω`)