1 files changed, 4 insertions, 3 deletions
diff --git a/tools/dataProvider.h b/tools/dataProvider.h
index 70a87f0..d598f3f 100644
--- a/tools/dataProvider.h
+++ b/tools/dataProvider.h
@@ -2,10 +2,11 @@
 #define DATAPROV_H
 
 #include<Eigen/Dense>
+#include "../tools/easylogging++.h"
 
 class DataProvider  //Virtual base class for data input
 {
-private:
+protected:
     int size;
     int attrSize;
 public:
@@ -17,9 +18,9 @@ public:
     }
 
     virtual Eigen::MatrixXd* getAttr() = 0;
-    virtual Eigen::MatrixXd* getPref() = 0;
+    virtual Eigen::VectorXd* getPref() = 0;
     virtual int open();
-    virtual int parse();
+    virtual bool EOFile();
 };
 
 #endif
 \ No newline at end of file