Python Matrix Factorization (PyMF) is a module for several constrained/unconstrained matrix factorization (and related) methods for both sparse and dense matrrices (for a brief introduction to factorization of gigantric matrices have a look at a tutorial we gave at ECML-PKDD 2011). It requires cvxopt, numpy, and scipy.
PyMF currently includes the following methods:
Given a dataset, most factorization methods try to minimize the Frobenius norm
| data - W*H| by finding a suitable set of basis vectors
W and coefficients H. The syntax for calling the various methods is quite similar. Usually, one has to submit a desired number of basis vectors and the maximum number of iterations. For example, applying NMF to a dataset data aiming at 2 basis vectors within 10 iterations works as follows:
>>> import pymf >>> import numpy as np >>> data = np.array([[1.0, 0.0, 2.0], [0.0, 1.0, 1.0]]) >>> nmf_mdl = pymf.NMF(data, num_bases=2, niter=10) >>> nmf_mdl.initialization() >>> nmf_mdl.factorize()
The basis vectors are now stored in
nmf_mdl.W, the coefficients in
nmf_mdl.H. To compute coefficients for an existing set of basis vectors simply copy W to nmf_mdl.W, and set compW to False:
>>> data = np.array([[1.5], [1.2]]) >>> W = np.array([[1.0, 0.0], [0.0, 1.0]]) >>> nmf_mdl = pymf.NMF(data, num_bases=2, niter=1, compW=False) >>> nmf_mdl.initialization() >>> nmf_mdl.W = W >>> nmf_mdl.factorize()
By changing pymf.NMF to e.g. pymf.AA or pymf.CNMF Archetypal Analysis or Convex-NMF can be applied. Some methods might allow other parameters, make sure to have a look at the corresponding
>>>help(pymf.AA) documentation. For example, CUR, CMD, and SVD are handled slightly differently, as they factorize into three submatrices which requires appropriate arguments for row and column sampling.
For handling larger datasets pymf supports hdf5 via h5py. Usage is straight forward as h5py allows to map large numpy matrices to disk. Thus, instead of passing data as a np.array, you can simply send the corresponding hdf5 table. The following example shows how to apply pymf to a random matrix that is entirely stored on disk. In this example the dataset does not have to fit into memory, the resulting low-rank factors
W,H have to.
>>> import h5py >>> import numpy as np >>> import pymf >>> >>> file = h5py.File('myfile.hdf5', 'w') >>> file['dataset'] = np.random.random((100,1000)) >>> sivm_mdl = pymf.SIVM(file['dataset'], num_bases=10) >>> sivm_mdl.factorize()
If the low-rank matrices
W,H also do not fit into memory, they can be initialized as a h5py matrix.
>>> import h5py >>> import numpy as np >>> import pymf >>> >>> file = h5py.File('myfile.hdf5', 'w') >>> file['dataset'] = np.random.random((100,1000)) >>> file['W'] = np.random.random((100,10)) >>> file['H'] = np.random.random((10,1000)) >>> sivm_mdl = pymf.SIVM(file['dataset'], num_bases=10) >>> sivm_mdl.W = file['W'] >>> sivm_mdl.H = file['H'] >>> sivm_mdl.factorize()
Please note that currently not all methods work well with hdf5. While they all accept hdf5 input matrices, they sometimes lead to very high memory consumption on intermediate computation steps. This is difficult to avoid unless we switch to a completely disk-based storage.