nmf.py

# https://github.com/scikit-learn/scikit-learn/blob/master/sklearn/decomposition/nmf.py
from sklearn.decomposition import NMF
import numpy
A = numpy.random.uniform(size=[40, 30])
nmf_model = NMF(n_components=3, init='random', random_state=0)
nmf_model.fit(A)
W = nmf_model.fit_transform(A)
H = nmf_model.components_
print(A.shape)
print(W.shape)
print(H.shape)
print(numpy.dot(W, H).shape)
print(A - numpy.dot(W, H))