update omniglot classification notebook

...ap_png/heatmapMNIST_Stochastic_ST_first_conv_binaryfc.png

...ts/heatmap_png/heatmapOmniglot_classif_NonBinaryNetfc.png

...s/heatmap_png/heatmapOmniglot_classif_NonBinaryNetfc2.png

...mapOmniglot_classif_Stochastic_ST_first_conv_binaryfc.png

...apOmniglot_classif_Stochastic_ST_first_conv_binaryfc2.png

utils/models.py

visualize/viz.py