summaryrefslogtreecommitdiff
path: root/tools/fileDataProvider.h
diff options
context:
space:
mode:
authorJoe Zhao <ztuowen@gmail.com>2015-04-11 17:46:19 +0800
committerJoe Zhao <ztuowen@gmail.com>2015-04-11 17:46:19 +0800
commit85a4962556b67d1cc0668ecb2fbb03b3b4dd6e7e (patch)
tree3059605ab0707b5530330dacdd9445b59bd73669 /tools/fileDataProvider.h
parent5d57accc3e1bc2b89c1e7753f7dbb40f3c8f575a (diff)
downloadranksvm-85a4962556b67d1cc0668ecb2fbb03b3b4dd6e7e.tar.gz
ranksvm-85a4962556b67d1cc0668ecb2fbb03b3b4dd6e7e.tar.bz2
ranksvm-85a4962556b67d1cc0668ecb2fbb03b3b4dd6e7e.zip
completed & tested, train & predict
Diffstat (limited to 'tools/fileDataProvider.h')
-rw-r--r--tools/fileDataProvider.h11
1 files changed, 9 insertions, 2 deletions
diff --git a/tools/fileDataProvider.h b/tools/fileDataProvider.h
index 6ccf28f..942ec9b 100644
--- a/tools/fileDataProvider.h
+++ b/tools/fileDataProvider.h
@@ -2,6 +2,7 @@
#define FDPROV_H
#include "dataProvider.h"
+#include "easylogging++.h"
#include <string>
#include <iostream>
#include <fstream>
@@ -18,11 +19,17 @@ public:
DataEntry* e;
out.clear();
int fsize;
- out.setfSize(fsize);
fin>>fsize;
+ LOG(INFO)<<"Feature size:"<<fsize;
+ out.setfSize(fsize);
while (!fin.eof()) {
- e= new DataEntry;
+ e = new DataEntry;
fin>>e->rank;
+ if (e->rank == 0)
+ {
+ delete e;
+ break;
+ }
fin>>e->qid;
e->feature.resize(fsize);
for (int i=0;i<fsize;++i) {