search

carpentries-incubator / deep-learning-intro

Learn Deep Learning with Python
https://carpentries-incubator.github.io/deep-learning-intro/
Other
30 stars 36 forks source link

plot network strcuture at the end of episode 2.4 #442

Open cmtso opened 8 months ago

cmtso commented 8 months ago

This is my first submission. I think it would nice to show a plot for the network structure for the visual learners out there, perhaps using the plot_model function.

https://www.tensorflow.org/api_docs/python/tf/keras/utils/plot_model

svenvanderburg commented 7 months ago

Great suggestion @cmtso! 🙏 We could add it to episode 2 when we show the model summary. This would add some dependencies (I think graphviz?) to the lesson I think? But we could also add a infobox where we show that this is possible (the instructor could demo on their machine, so we don't bother participants with installing an extra package for just this visualization).

If you want you can give it a shot and open a PR, otherwise we will pick it up somewhere in the coming months when we have our next sprint.