seriesClassification.h

Go to the documentation of this file.

#ifndef seriesClassification_hpp
#define seriesClassification_hpp

#include <vector>
#include <string>
#include <map>
#include "fastDTW.h"
#include "trainingExample.h"

template<typename T>
class seriesClassificationTemplate {
public:
    
    seriesClassificationTemplate();
    ~seriesClassificationTemplate();
    
    bool train(const std::vector<trainingSeriesTemplate<T> > &seriesSet);
    
    void reset();
    
    std::string run(const std::vector<std::vector<T> > &inputSeries);
    
    T run(const std::vector<std::vector<T> > &inputSeries, std::string label);
    
    std::string runContinuous(const std::vector<T> &inputVector);
    
    std::vector<T> getCosts() const;
    
    int getMinLength() const;
    
    int getMinLength(std::string label) const;
    
    int getMaxLength() const;
    
    int getMaxLength(std::string label) const;
    
    template<typename TT>
    struct minMax {
        TT min;
        TT max;
    };
    
    minMax<T> calculateCosts(std::string label) const;
    
    minMax<T> calculateCosts(std::string label1, std::string label2) const;
    
private:
    std::vector<trainingSeriesTemplate<T> > allTrainingSeries;
    int vectorLength;
    std::vector<T> allCosts;
    int maxLength;
    int minLength;
    std::map<std::string, minMax<int> > lengthsPerLabel;
    
    std::vector<std::vector<T> > seriesBuffer;
    int hopSize;
    int counter;
};

//This is here to keep the old API working
using seriesClassification = seriesClassificationTemplate<double>;
using seriesClassificationFloat = seriesClassificationTemplate<float>;

#endif