def build_nmf(X, k=5): mod = NMF(n_components=k) W = mod.fit_transform(X) H = mod.components_ return W, H