/**
 *  Dataset class header file. 
 *
 *  Vladimir Vacic
 *  Computer Science and Engineering Department
 *  University of California, Riverside
 *
 *  May-6-2005 
 *
 */

#ifndef _DATASET_H_
#define _DATASET_H_

#include <iostream>
#include <vector>
using namespace std;


class Dataset    {

public:

    Dataset();
    Dataset(const vector<vector<double> > &tdata, const vector<double> &tlabels);
    Dataset(const Dataset &tset);
    ~Dataset();

    void load(char *filename);

    /** Returns number of features. */
    int width() const;

    /** Returns number of examples. */
    int length() const;

    double         operator()(int r, int c) const;
    vector<double> column(int c) const;
    vector<double> row(int r) const;

    vector<vector<double> > getData() const;
    vector<double> getLabels() const;
    
    /** Z-score normalizes the data set, column by column. */
    void normalize();

private:

    vector<vector<double> > data;
    vector<double>          labels;
};

#endif