INNER CODE UNIT · Python

NMF

ShixiangWang/sigminer · inst/py/nmf.py:13

def NMF(V, rank):
  '''Run NMF for a target matrix.
  V: the target matrix to decompose.'''
  nmf = nimfa.Nmf(
    V, 
    seed="random_c", 
    rank=rank,
    n_run=1,
    max_iter=2000,
    update='divergence',
    objective='div')
  nmf_fit = nmf()
  W = nmf_fit.basis()
  K = W.shape[1]
  print('Stop at iteration #: %d' % nmf_fit.summary()["n_iter"])
  return {"W":W, "H":nmf_fit.coef(), "K":K, "KLD":nmf_fit.distance(metric='kl')}

def NMF2(i, V, rank):

View source record →