plot samples

Author: aamini

mitdeeplearning/util.py

@@ -9,32 +9,6 @@
 
 
 
-#####################################
-def custom_progress_text(message):
-
-  message_ = message.replace('(', '{')
-  message_ = message_.replace(')', '}')
-
-  keys = [key[1] for key in Formatter().parse(message_)]
-
-  ids = {}
-  for key in keys:
-    if key is not None:
-      ids[key] = float('nan')
-
-  msg = progressbar.FormatCustomText(message, ids)
-  return msg
-
-def create_progress_bar(text=None):
-  if text is None:
-    text = progressbar.FormatCustomText('')
-  bar = progressbar.ProgressBar(widgets=[
-      progressbar.Percentage(),
-      progressbar.Bar(),
-      progressbar.AdaptiveETA(), '  ',
-      text,
-  ])
-  return bar
 
 def display_model(model):
   tf.keras.utils.plot_model(model,
@@ -43,6 +17,22 @@ def display_model(model):
   return ipythondisplay.Image('tmp.png')
 
 
+def plot_sample(x,y,vae):
+    plt.figure(figsize=(2,1))
+    plt.subplot(1, 2, 1)
+
+    idx = np.where(y.numpy()==1)[0][0]
+    plt.imshow(x[idx])
+    plt.grid(False)
+
+    plt.subplot(1, 2, 2)
+    _, _, _, recon = vae(x)
+    plt.imshow(recon[idx])
+    plt.grid(False)
+
+    plt.show()
+
+
 class LossHistory:
   def __init__(self, smoothing_factor=0.0):
     self.alpha = smoothing_factor

setup.py

@@ -21,13 +21,13 @@ def get_dist(pkgname):
 setup(
   name = 'mitdeeplearning',         # How you named your package folder (MyLib)
   packages = ['mitdeeplearning'],   # Chose the same as "name"
-  version = '0.4.4',      # Start with a small number and increase it with every change you make
+  version = '0.4.5',      # Start with a small number and increase it with every change you make
   license='MIT',        # Chose a license from here: https://help.github.com/articles/licensing-a-repository
   description = 'Official software labs for MIT Introduction to Deep Learning (http://introtodeeplearning.com)',   # Give a short description about your library
   author = 'Alexander Amini',                   # Type in your name
   author_email = '[email protected]',      # Type in your E-Mail
   url = 'http://introtodeeplearning.com',   # Provide either the link to your github or to your website
-  download_url = 'https://github.com/aamini/introtodeeplearning_labs/archive/v0.4.4.tar.gz',    # I explain this later on
+  download_url = 'https://github.com/aamini/introtodeeplearning_labs/archive/v0.4.5.tar.gz',    # I explain this later on
   keywords = ['deep learning', 'neural networks', 'tensorflow', 'introduction'],   # Keywords that define your package best
   install_requires=install_deps,
   classifiers=[