comparison cnmf.py @ 7:294f66d285af

some previous changes

 import pymf
 
 # Local stuff
 from utils import SegUtil
 
+# Algorithm params
+h = 8               # Size of median filter for features in C-NMF
+R = 15              # Size of the median filter for the activation matrix C-NMF
+rank = 4            # Rank of decomposition for the boundaries
+rank_labels = 6     # Rank of decomposition for the labels
+R_labels = 6        # Size of the median filter for the labels
 
 def cnmf(S, rank, niter=500):
     """(Convex) Non-Negative Matrix Factorization.
 
     Parameters

     G = np.sum(G, axis=1)
     G = SegUtil.median_filter(G[:, np.newaxis], R)
     return G.flatten()
 
 
-def segmentation(X, rank, R, h, niter=300):
+def segmentation(X, rank=4, R=15, h=8, niter=300):
     """
     Gets the segmentation (boundaries and labels) from the factorization
     matrices.
 
     Parameters

             rank += 1
             bound_idxs = None
         else:
             break
 
-    return bound_idxs
+    return G, bound_idxs