[PYTHON] Lassen Sie uns GraphConvModel von DeepChem visualisieren

Einführung

Ich wollte Deep Learning mit einer Verbindung starten, also habe ich das GraphConvModel von DeepChem das letzte Mal gehackt und es mit der Zusammenfassungsmethode als Text ausgegeben.

Diesmal habe ich versucht, es mit plot_model zu visualisieren.

Umgebung

• DeepChem 2.3
• pydot

Methode

Wenn Sie den folgenden Code in Zeile 624 der Datei einfügen, in der die GraphConvModel-Klassendefinition erstellt wird, und ein Vorhersagemodell mit den entsprechenden Daten erstellen, wird eine Datei mit dem Namen model.png generiert.

<anaconda3>/envs/deepchem/lib/python3.7/site-packages/deepchem/models/graph_conv.py

    from tensorflow.keras.utils import plot_model
    plot_model(
            model,
            show_shapes=True,
        )

Ergebnis

So was. Die Reihenfolge der Eingabe wird geändert und ist schwer zu erkennen, aber möglicherweise leichter vorstellbar als die textbasierte. Das Weben wird komplizierter.