branched for mem

1 files changed, 1 insertions, 3 deletions

diff --git a/train.cpp b/train.cpp
index a0c62a9..05787c9 100644
--- a/train.cpp
+++ b/train.cpp
@@ -22,14 +22,12 @@ int train(DataProvider &dp) {
     rsvm = RSVM::loadModel(vm["model"].as<string>());
 
     dp.open();
-    DataList D;
+    RidList D;
 
     LOG(INFO)<<"Training started";
     dp.getAllDataSet(D);
     LOG(INFO)<<"Read "<<D.getSize()<<" entries with "<< D.getfSize()<<" features";
     rsvm->train(D);
-    vector<double> L;
-    rsvm->predict(D,L);
 
     LOG(INFO)<<"Training finished,saving model";