small display and comment changes

example_multivariate.py

@@ -1,4 +1,4 @@
-"""Dictionary recovering experiment for univariate random dataset"""
+"""Dictionary recovering experiment for multivariate random dataset"""
 import numpy as np
 import matplotlib.pyplot as plt
 from mdla import MultivariateDictLearning, MiniBatchMultivariateDictLearning
@@ -8,9 +8,6 @@
 from numpy import array, arange, zeros, min, max
 from numpy.random import rand, randn, permutation, randint
 
-# TODO: Add SNR, repeat experiments to make stats, make a fast and a 
-#       long version, 
-
 def plot_multivariate(objective_error, detection_rate, wasserstein,
                     n_iter, figname):
     fig = plt.figure(figsize=(15,5))

example_univariate.py

@@ -8,9 +8,6 @@
 from numpy import array, arange, zeros, min, max
 from numpy.random import rand, randn, permutation, randint
 
-# TODO: Add SNR, repeat experiments to make stats, make a fast and a 
-#       long version, 
-
 def plot_univariate(objective_error, detection_rate, wasserstein,
                     n_iter, figname):
     fig = plt.figure(figsize=(15,5))

mdla.py

@@ -1002,11 +1002,10 @@ def multivariate_dict_learning_online(X, n_kernels=2, n_nonzero_coefs=1,
             if ii == (iter_offset+1)*int(n_batches) and verbose == 1:
                 print ('Expecting this learning iterations to finish in',
                        (time()-t0)*n_iter/60., 'm')
-            # if verbose == 1:
+            if verbose == 1:
                 print ('Time from begining is', time()-t0, 's, with n_iter=',
                         n_iter, ', iter_offset=', iter_offset,
-                        ', i.e.', n_iter, 'iterations to go. ii=',
-                        ii)
+                        ', i.e.', n_iter, 'iterations to go.')
         except KeyboardInterrupt:
             break