summaryrefslogtreecommitdiff
path: root/tools
diff options
context:
space:
mode:
authorJoe Zhao <ztuowen@gmail.com>2015-03-11 00:55:41 +0800
committerJoe Zhao <ztuowen@gmail.com>2015-03-11 00:55:41 +0800
commitf6c22b46449fa77f90e319e4b159ccb6c2a5732b (patch)
treebc38dde03ffa5cb3c2fea9f5fefff0b990de405b /tools
parent3d204f5fe4614624ca342090feecbfe4df188d9d (diff)
downloadranksvm-f6c22b46449fa77f90e319e4b159ccb6c2a5732b.tar.gz
ranksvm-f6c22b46449fa77f90e319e4b159ccb6c2a5732b.tar.bz2
ranksvm-f6c22b46449fa77f90e319e4b159ccb6c2a5732b.zip
restructure, changed label type
Diffstat (limited to 'tools')
-rw-r--r--tools/dataProvider.h3
-rw-r--r--tools/fileDataProvider.h1
2 files changed, 3 insertions, 1 deletions
diff --git a/tools/dataProvider.h b/tools/dataProvider.h
index ce2bf12..d311149 100644
--- a/tools/dataProvider.h
+++ b/tools/dataProvider.h
@@ -17,7 +17,7 @@
typedef Eigen::MatrixXd DataSet;
-typedef std::vector<double> Labels;
+typedef Eigen::VectorXd Labels;
class DataProvider //Virtual base class for data input
{
@@ -39,6 +39,7 @@ public:
virtual int getDataSet(DataSet &out) = 0;
virtual int getLabel(Labels &out) = 0;
virtual int open()=0;
+ virtual int close()=0;
};
#endif \ No newline at end of file
diff --git a/tools/fileDataProvider.h b/tools/fileDataProvider.h
index 8a499ca..a4cc252 100644
--- a/tools/fileDataProvider.h
+++ b/tools/fileDataProvider.h
@@ -18,6 +18,7 @@ public:
return 0;
};
virtual int open(){eof=true;return 0;};
+ virtual int close(){return 0;};
};
#endif \ No newline at end of file