mog_em_main.cc

Go to the documentation of this file.

/* MLPACK 0.2
 *
 * Copyright (c) 2008, 2009 Alexander Gray,
 *                          Garry Boyer,
 *                          Ryan Riegel,
 *                          Nikolaos Vasiloglou,
 *                          Dongryeol Lee,
 *                          Chip Mappus, 
 *                          Nishant Mehta,
 *                          Hua Ouyang,
 *                          Parikshit Ram,
 *                          Long Tran,
 *                          Wee Chin Wong
 *
 * Copyright (c) 2008, 2009 Georgia Institute of Technology
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 */
#include "mog_em.h"

const fx_entry_doc mog_em_main_entries[] = {
  {"data", FX_REQUIRED, FX_STR, NULL,
   " A file containing the data on which the model"
   " has to be fit.\n"},
  {"output", FX_PARAM, FX_STR, NULL,
   " The file into which the output is to be written into.\n"},
  FX_ENTRY_DOC_DONE
};

const fx_submodule_doc mog_em_main_submodules[] = {
  {"mog_em", &mog_em_doc,
   " Responsible for intializing the model and"
   " computing the parameters.\n"},
  FX_SUBMODULE_DOC_DONE
};

const fx_module_doc mog_em_main_doc = {
  mog_em_main_entries, mog_em_main_submodules,
  " This program test drives the parametric estimation "
  "of a Gaussian mixture model using maximum likelihood.\n"
};

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

  fx_module *root = 
    fx_init(argc, argv, &mog_em_main_doc);

  
  const char *data_filename = fx_param_str_req(root, "data");

  Matrix data_points;
  data::Load(data_filename, &data_points);


  MoGEM mog;

  struct datanode* mog_em_module = 
    fx_submodule(root, "mog_em");
  fx_param_int(mog_em_module, "K", 1);
  fx_set_param_int(mog_em_module, "D", data_points.n_rows());

  fx_timer_start(mog_em_module, "model_init");
  mog.Init(mog_em_module);
  fx_timer_stop(mog_em_module, "model_init");

  ArrayList<double> results;

  fx_timer_start(mog_em_module, "EM");
  mog.ExpectationMaximization(data_points);
  fx_timer_stop(mog_em_module, "EM");
  
  mog.Display();
  mog.OutputResults(&results);


  const char *output_filename = fx_param_str(NULL, "output", "output.csv");

  FILE *output_file = fopen(output_filename, "w");

  ot::Print(results, output_file);
  fclose(output_file);
  fx_done(root);

  return 1;
}