Comment stocker la visualisation graphique MXNet en tant qu'image dans un fichier?

Question

Selon la page this, le graphique de calcul MXNet peut être visualisé en utilisant mx.viz.plot_network(net).

Mais cela ne fonctionne que sur les ordinateurs portables Jupyter. Comment puis-je le faire sans ordinateur portable Jupyter? Est-il possible de sauvegarder la visualisation sous forme d'image dans un fichier?

Answer 1

mx.viz.plot_network renvoie un objet graphviz.dot.Digraph et peut être rendu comme tout autre objet Digraph.

Voici un exemple:

# Store the Digraph in a variable 
graph = mx.viz.plot_network(net) 

# Pick an image format from this list: http://www.graphviz.org/doc/info/output.html 
graph.format = 'png' 

# Choose a filename and render the image. 
graph.render('graph') 

code ci-dessus va rendre le graphique 'graph.png' dans le répertoire courant.