trainingData.h

Go to the documentation of this file.

#ifndef trainingData_h
#define trainingData_h

#include <vector>
#include <string>
#include <unordered_map>

#include "rapidMix.h"
#include "json.h"

namespace rapidmix {
    
    class trainingData {
        
    public:
        
        trainingData();
        
        struct element{
            uint32_t uniqueId; //MZ: Does this scope of this id need to extend beyond this instantiation?
            std::vector<double> input;
            std::vector<double> output;
            double timeStamp;
        };
        
        struct phrase {
            uint32_t uniqueId;
            std::string label; //TODO: Need to work this with templates
            std::vector<std::string> columnNames;
            std::vector<element> elements;
        };
        
        std::vector<phrase> trainingSet;
        
        uint32_t startRecording(); //FIXME: this should go away. -MZ
        
        uint32_t startRecording(const std::string &label);
        
        uint32_t addElement(const std::vector<double> &input, const std::vector<double> &output);
        
        uint32_t addElement(const std::vector<double> &input);
        
        void stopRecording();
        
        uint32_t recordSingleElement(const std::string &label, const std::vector<double> &input);
        
        uint32_t recordSingleElement(const std::vector<double> &input, const std::vector<double> &output);
        
        uint32_t recordSingleElement(const std::string &label, const std::vector<double> &input, const std::vector<double> &output);
        
        std::vector<std::string> getColumnNames();
        void setColumnNames(const std::vector<std::string> &columnNames);
        
        
        std::string getJSON();
        void writeJSON(const std::string &filepath);
        bool putJSON(const std::string &jsonMessage);
        bool readJSON(const std::string &filepath);
        
    private:
        int targetPhrase;
        uint32_t currentId;
        
        //* Returns and increments current id */
        uint32_t assignCurrentId();
        
        Json::Value parse2json();
        void json2trainingSet(const Json::Value &newTrainingData);
        
    };
    
}

#endif