1 files changed, 3 insertions, 7 deletions
diff --git a/main.cpp b/main.cpp
index 0c71b07..a517d88 100644
--- a/main.cpp
+++ b/main.cpp
@@ -19,7 +19,6 @@ int train() {
     rsvm = RSVM::loadModel(vm["model"].as<std::string>());
     FileDP dp(vm["feature"].as<std::string>());
 
-    // Generic training operations
     dp.open();
     DataList D;
 
@@ -50,12 +49,9 @@ int predict() {
     std::vector<double> L;
     LOG(INFO)<<"Prediction started";
 
-    while (!dp.EOFile())
-    {
-        dp.getDataSet(D);
-        LOG(INFO)<<"Read "<<D.getSize()<<" entries with "<< D.getfSize()<<" features";
-        rsvm->predict(D,L);
-    }
+    dp.getDataSet(D);
+    LOG(INFO)<<"Read "<<D.getSize()<<" entries with "<< D.getfSize()<<" features";
+    rsvm->predict(D,L);
 
     if (vm.count("validate"))
     {