modelSet.h

Go to the documentation of this file.

 
#ifndef MODELSET_H
#define MODELSET_H
 
#include <vector>
#include "trainingExample.h"
#include "baseModel.h"
#include "neuralNetwork.h"
#include "knnClassification.h"
#include "svmClassification.h"
#ifndef EMSCRIPTEN
#include "../dependencies/json/json.h"
#endif
 
template<typename T>
class modelSet {
public:
    modelSet();
    virtual ~modelSet();
    virtual bool train(const std::vector<trainingExampleTemplate<T> > &trainingSet);
    bool reset();
    
    std::vector<T> run(const std::vector<T> &inputVector);
    
protected:
    std::vector<baseModel<T>*> models {};
    int numInputs {};
    std::vector<std::string> inputNames {};
    int numOutputs {};
    bool isTraining {}; //This is true while the models are training, and will block running
    bool isTrained {};
 
    void threadTrain(std::size_t i, const std::vector<trainingExampleTemplate<T> >& training_set);
 
#ifndef EMSCRIPTEN //The javascript code will do its own JSON parsing
public:
 
    std::string getJSON();
 
    void writeJSON(const std::string &filepath);
 
    bool putJSON(const std::string &jsonMessage);
    bool readJSON(const std::string &filepath);
        
private:
    Json::Value parse2json();
    void json2modelSet(const Json::Value &root);
#endif
};
 
#endif