traintestsets.h

Go to the documentation of this file.

#ifndef TRAINTESTSETS_H
#define TRAINTESTSETS_H

/*
 * traintestsets.h
 * Copyright (c) 2016 Georgetown University.  All Rights Reserved.
 */

#include <iostream>
#include "dataset.h"

using namespace std;

class TrainTestSets {

  friend ostream &operator<<( ostream &out, const TrainTestSets &tts );

  protected:

    DataSet train;

    DataSet test;

  public:

    TrainTestSets();

    TrainTestSets( const vector<string> &options ) throw ( logic_error );

    DataSet getTrainingSet() const;

    DataSet getTestingSet() const;

    void setTrainingSet( const DataSet &train );

    void setTestingSet( const DataSet &test );

    void setOptions( const vector<string> &options ) throw ( logic_error );

}; // TrainTestSets class

#endif