1 files changed, 15 insertions, 2 deletions
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;
 }