diff options
| author | Joe Zhao <ztuowen@gmail.com> | 2015-05-22 11:31:25 +0800 |
|---|---|---|
| committer | Joe Zhao <ztuowen@gmail.com> | 2015-05-22 11:31:25 +0800 |
| commit | 01ea2597d922774ff641d6a2eacae22979d68802 (patch) | |
| tree | e59713ad85311938f05ab5ad28c82374a3005e72 /tools/fileDataProvider.h | |
| parent | 482be9136a726923eb24c9f5c0244e94c0fc91bc (diff) | |
| download | ranksvm-01ea2597d922774ff641d6a2eacae22979d68802.tar.gz ranksvm-01ea2597d922774ff641d6a2eacae22979d68802.tar.bz2 ranksvm-01ea2597d922774ff641d6a2eacae22979d68802.zip | |
added shuffle option
Diffstat (limited to 'tools/fileDataProvider.h')
| -rw-r--r-- | tools/fileDataProvider.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tools/fileDataProvider.h b/tools/fileDataProvider.h index 567c8e2..972a4c5 100644 --- a/tools/fileDataProvider.h +++ b/tools/fileDataProvider.h @@ -38,11 +38,20 @@ public: void readEntries(); int getfSize() { if(!read) readEntries(); return d.getfSize();}; int getpSize(); + void shuffle(std::vector<DataEntry*> &dat); virtual void getAllDataSet(RidList &out); virtual int getDataSet(DataList &out); virtual int open(){fin.open(fname); eof=false;return 0;}; virtual int close(){fin.close(); d.clear();return 0;}; void take(int n,std::vector<DataEntry*> &a,std::vector<DataEntry*> &b); + void getRidVector(std::vector<DataEntry*> &rid){ + if (!read) + readEntries(); + rid.clear(); + std::vector<DataEntry*> &dat = d.getData(); + for (int i=0;i<dat.size();++i) + rid.push_back(dat[i]); + } }; #endif
\ No newline at end of file |