modelSet.h

Go to the documentation of this file.

//
//  modelSet.h
//  RapidLib
//
//  Created by mzed on 26/09/2016.
//  Copyright © 2016 Goldsmiths. All rights reserved.
//

#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>*> myModelSet;
    int numInputs;
    std::vector<std::string> inputNames;
    int numOutputs;
    bool created;

#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