nmf.cpp

Go to the documentation of this file.

/* Copyright 2016 Ramakrishnan Kannan */

#include "common/nmf.hpp"
#include <stdio.h>
#include <string>
#include "common/parsecommandline.hpp"
#include "common/utils.hpp"
#include "nmf/aoadmm.hpp"
#include "nmf/bppnmf.hpp"
#include "nmf/hals.hpp"
#include "nmf/mu.hpp"

template <class NMFTYPE>
void NMFDriver(int k, UWORD m, UWORD n, std::string AfileName,
               std::string WfileName, std::string HfileName, int numIt) {
#ifdef BUILD_SPARSE
  SP_MAT A;
#else
  MAT A;
#endif
  double t2;
  if (!AfileName.empty() &&
      !AfileName.compare(AfileName.size() - 4, 4, "rand")) {
#ifdef BUILD_SPARSE
    A.load(AfileName, arma::coord_ascii);
    INFO << "Successfully loaded the input matrix" << std::endl;
#else
    tic();
    A.load(AfileName);
    t2 = toc();
    INFO << "Successfully loaded dense input matrix. A=" << PRINTMATINFO(A)
         << " took=" << t2 << std::endl;
    m = A.n_rows;
    n = A.n_cols;
#endif
  } else {
#ifdef BUILD_SPARSE
    A = arma::sprandu<SP_MAT>(m, n, 0.001);
#else
    A = arma::randu<MAT>(m, n);
#endif
    INFO << "generated random matrix A=" << PRINTMATINFO(A) << std::endl;
  }
  MAT W, H;
  NMFTYPE nmfAlgorithm(A, k);
  nmfAlgorithm.num_iterations(numIt);
  INFO << "completed constructor" << PRINTMATINFO(A) << std::endl;
  tic();
  nmfAlgorithm.computeNMF();
  t2 = toc();
  INFO << "time taken:" << t2 << std::endl;
  if (WfileName.compare("_w") != 0) {
    nmfAlgorithm.getLeftLowRankFactor().save(WfileName, arma::raw_ascii);
  }
  if (HfileName.compare("_h") != 0) {
    nmfAlgorithm.getRightLowRankFactor().save(HfileName, arma::raw_ascii);
  }
}
#ifdef BUILD_SPARSE
void incrementalGraph(std::string AfileName, std::string WfileName) {
  SP_MAT A;  
  A.load(AfileName);
  INFO << "Loaded input matrix A=" << PRINTMATINFO(A) << std::endl;
  MAT W, H;
  W.load(WfileName);
  INFO << "Loaded input matrix W=" << PRINTMATINFO(W) << std::endl;
  H.ones(A.n_cols, W.n_cols);
  planc::BPPNMF<SP_MAT> bppnmf(A, W, H);
  H = bppnmf.solveScalableNNLS();
  OUTPUT << H << std::endl;
}
#endif
void parseCommandLineandCallNMF(int argc, char* argv[]) {
  planc::ParseCommandLine pc(argc, argv);
  pc.parseplancopts();
  pc.printConfig();
  switch (pc.lucalgo()) {
    case MU:
#ifdef BUILD_SPARSE
      NMFDriver<planc::MUNMF<SP_MAT> >(
          pc.lowrankk(), pc.globalm(), pc.globaln(), pc.input_file_name(),
          pc.output_file_name() + "_w", pc.output_file_name() + "_h",
          pc.iterations());
#else
      NMFDriver<planc::MUNMF<MAT> >(
          pc.lowrankk(), pc.globalm(), pc.globaln(), pc.input_file_name(),
          pc.output_file_name() + "_w", pc.output_file_name() + "_h",
          pc.iterations());
#endif
      break;
    case HALS:
#ifdef BUILD_SPARSE
      NMFDriver<planc::HALSNMF<SP_MAT> >(
          pc.lowrankk(), pc.globalm(), pc.globaln(), pc.input_file_name(),
          pc.output_file_name() + "_w", pc.output_file_name() + "_h",
          pc.iterations());
#else
      NMFDriver<planc::HALSNMF<MAT> >(
          pc.lowrankk(), pc.globalm(), pc.globaln(), pc.input_file_name(),
          pc.output_file_name() + "_w", pc.output_file_name() + "_h",
          pc.iterations());
#endif
      break;
    case ANLSBPP:
#ifdef BUILD_SPARSE
      NMFDriver<planc::BPPNMF<SP_MAT> >(
          pc.lowrankk(), pc.globalm(), pc.globaln(), pc.input_file_name(),
          pc.output_file_name() + "_w", pc.output_file_name() + "_h",
          pc.iterations());
#else
      NMFDriver<planc::BPPNMF<MAT> >(
          pc.lowrankk(), pc.globalm(), pc.globaln(), pc.input_file_name(),
          pc.output_file_name() + "_w", pc.output_file_name() + "_h",
          pc.iterations());
#endif
      break;
    case AOADMM:
#ifdef BUILD_SPARSE
      // NMFDriver<AOADMMNMF<SP_MAT > >(lowRank, m, n, AfileName, WInitfileName,
      //                                HInitfileName, WfileName, HfileName,
      //                                numIt);
      NMFDriver<planc::AOADMMNMF<SP_MAT> >(
          pc.lowrankk(), pc.globalm(), pc.globaln(), pc.input_file_name(),
          pc.output_file_name() + "_w", pc.output_file_name() + "_h",
          pc.iterations());

#else
      NMFDriver<planc::AOADMMNMF<MAT> >(
          pc.lowrankk(), pc.globalm(), pc.globaln(), pc.input_file_name(),
          pc.output_file_name() + "_w", pc.output_file_name() + "_h",
          pc.iterations());
#endif
      break;
    default:
      ERR << "Not a valid algorithm" << std::endl;
  }
}

int main(int argc, char* argv[]) { parseCommandLineandCallNMF(argc, argv); }