baseModel.h

Go to the documentation of this file.

 
#ifndef BASEMODEL_H
#define BASEMODEL_H
 
#include <vector>
#include "trainingExample.h"
 
#ifndef EMSCRIPTEN
#include "../dependencies/json/json.h"
#endif
 
#ifdef __ANDROID__
#include <ostream>
namespace std
{ // Workaround for android
    template <typename T>
    std::string to_string(T Value)
    {
        std::ostringstream TempStream;
        TempStream << Value;
        return TempStream.str();
    }
 
    inline long double strtold(const char* str, char** str_end)
    {
        return strtod(str, str_end);
    }
}
#endif
 
template<typename T>
class baseModel
{
public:
    virtual ~baseModel() {};
    virtual T run(const std::vector<T>& inputVector) = 0; //TODO: I'd like this to be const
    virtual void train(const std::vector<trainingExampleTemplate<T> >& trainingSet) = 0;
    virtual void train(const std::vector<trainingExampleTemplate<T> >& trainingSet, const std::size_t whichOutput) = 0;
    virtual void reset() = 0;;
    virtual size_t getNumInputs() const = 0;
    virtual std::vector<size_t> getWhichInputs() const = 0;
 
#ifndef EMSCRIPTEN
    virtual void getJSONDescription(Json::Value& currentModel) = 0;
 
protected:
    
    template<typename TT, class Dummy=int>
    Json::Value vector2json(TT vec) 
    {
        Json::Value toReturn; 
        for (auto value : vec)
        {
            toReturn.append( (Json::Value)value );
        }
        return toReturn;
    }
 
    //FIXME: This is a temporary hack because Json::Value doesn't know what to do with unsinged longs, and XCode cares
    template<class Dummy=int>
    Json::Value vector2json(std::vector<unsigned long> vec)
    {
        Json::Value toReturn;
        for (auto value : vec)
        {
            toReturn.append( (double)value ); //I chose double here because that's close to what JS uses
        }
        return toReturn;
    }
#endif
 
};
 
#endif