1 files changed, 38 insertions, 2 deletions
diff --git a/main.cpp b/main.cpp
index 4765dcc..d23ded2 100644
--- a/main.cpp
+++ b/main.cpp
@@ -1,18 +1,54 @@
 #include <iostream>
 #include <Eigen/Dense>
+#include <boost/program_options.hpp>
+#include <string>
 #include "tools/easylogging++.h"
+#include "model/ranksvmtn.h"
+#include "tools/fileDataProvider.h"
 
 INITIALIZE_EASYLOGGINGPP
 
 using Eigen::MatrixXd;
+namespace po = boost::program_options;
 
-int main()
+po::variables_map vm;
+
+int train()
+{
+  RSVM* rsvm;
+  rsvm=RSVM::loadModel(vm["model"].as<std::string>().c_str());
+  FileDP dp(vm["feature"].as<std::string>().c_str());
+  rsvm->train(dp);
+  rsvm->predict(dp);
+  rsvm->saveModel(vm["output"].as<std::string>().c_str());
+  delete rsvm;
+  return 0;
+}
+
+int main(int argc,char ** argv)
 {
+  // Defining program options
+  po::options_description desc("Allowed options");
+  desc.add_options()
+          ("help,h", "produce help message")
+          ("model,m", po::value<std::string>()->default_value("test.m"), "set input model file")
+          ("feature,i", po::value<std::string>()->default_value("test.f"), "set input feature file")
+          ("output,o", po::value<std::string>()->default_value("test.m.out"), "set output model file")
+          ;
+
+  // Parsing program options
+  po::store(po::parse_command_line(argc, argv, desc), vm);
+  po::notify(vm);
+
+  if (vm.count("help")) {
+    std::cout << desc << "\n";
+    return 1;
+  }
+
   MatrixXd m(2,2);
   m(0,0) = 3;
   m(1,0) = 2.5;
   m(0,1) = -1;
   m(1,1) = m(1,0) + m(0,1);
-  LOG(FATAL) << "My first info log using default logger";
   std::cout << m << std::endl;
 }