[PYTHON] Visualisons GraphConvModel de DeepChem

introduction

Je voulais commencer le Deep Learning avec un composé, j'ai donc piraté le GraphConvModel de DeepChem la dernière fois et l'ai sorti sous forme de texte par la méthode de résumé.

Cette fois, j'ai essayé de le visualiser avec plot_model.

environnement

• DeepChem 2.3
• pydot

Méthode

Si vous mettez le code suivant à la ligne 624 du fichier où la classe GraphConvModel est définie et créez un modèle de prédiction avec les données appropriées, un fichier appelé model.png sera généré.

<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,
        )

résultat

Comme ça. L'ordre des entrées est modifié et il est difficile à voir, mais il peut être plus facile à imaginer que celui basé sur du texte. Le tissage va être plus compliqué.