---

TrainingSet.h

Go to the documentation of this file.

#ifndef _TRAININGSET_H
#define _TRAININGSET_H

#include "Neuron.h"
#include "Network.h"

namespace annie
{

class TrainingSet
{
protected:
        std::vector< VECTOR > _inputs;

        std::vector< VECTOR > _outputs;

        std::vector< VECTOR >::iterator _inputIter;

        std::vector< VECTOR >::iterator _outputIter;

        int _nInputs;

        int _nOutputs;

        void save_binary(const char *filename);

        void save_text(const char *filename);

        void load_binary(const char *filename);

        void load_text(const char *filename);
public:
        TrainingSet(int in,int out);

        TrainingSet(const char *filename,int file_type = TEXT_FILE);

        virtual ~TrainingSet();

        void addIOpair(real *input,real *output);

        void addIOpair(VECTOR input,VECTOR output);

        virtual void initialize();

        virtual bool epochOver();

        virtual int getSize();

        virtual int getInputSize();

        virtual int getOutputSize();

        virtual void getNextPair(VECTOR &input, VECTOR &desired);

        friend std::ostream& operator << (std::ostream& s, TrainingSet &T);
        
        virtual void save(const char *filename, int file_type = TEXT_FILE);

        virtual const char *getClassName();
};

}; //namespace annie
#endif // define _TRAININGSET_H

---