From f2d01e30f459818f0589e06839d38999aecfdc06 Mon Sep 17 00:00:00 2001
From: Joe Zhao <ztuowen@gmail.com>
Date: Sun, 8 Mar 2015 17:47:33 +0800
Subject: scaffolding

---
 main.cpp | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

(limited to 'main.cpp')

diff --git a/main.cpp b/main.cpp
index 8d5b393..87f9ce5 100644
--- a/main.cpp
+++ b/main.cpp
@@ -17,7 +17,14 @@ 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);
+    DataSet D;
+    Labels L;
+    while (!dp.EOFile())
+    {
+        dp.getDataSet(D);
+        dp.getLabel(L);
+        rsvm->train(D,L);
+    }
     rsvm->saveModel(vm["output"].as<std::string>().c_str());
     delete rsvm;
     return 0;
@@ -27,7 +34,13 @@ int predict() {
     RSVM *rsvm;
     rsvm = RSVM::loadModel(vm["model"].as<std::string>().c_str());
     FileDP dp(vm["feature"].as<std::string>().c_str());
-    rsvm->predict(dp);
+    DataSet D;
+    MatrixXd L;
+    while (!dp.EOFile())
+    {
+        dp.getDataSet(D);
+        rsvm->predict(D,L);
+    }
     delete rsvm;
     return 0;
 }
-- 
cgit v1.2.3-70-g09d2