summaryrefslogtreecommitdiff
path: root/tools/dataProvider.h
blob: d9440ce4084823060735a53fd815397bb61a1012 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#ifndef DATAPROV_H
#define DATAPROV_H

#include<Eigen/Dense>
#include "../tools/easylogging++.h"

// TODO decide how to construct training data

class DataProvider  //Virtual base class for data input
{
protected:
    int size;
    int attrSize;
public:
    int getSize(){
        return size;
    }
    int getAttrSize(){
        return attrSize;
    }

    virtual Eigen::MatrixXd* getAttr() = 0;
    virtual Eigen::VectorXd* getPref() = 0;
    virtual int open()=0;
    virtual bool EOFile()=0;
};

#endif