classification.h

Go to the documentation of this file.

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

#ifndef classification_h
#define classification_h

#include <vector>
#include "modelSet.h"

template<typename T>
class classificationTemplate : public modelSet<T> {
public:
    enum classificationTypes { knn, svm };
    
    classificationTemplate();
    
    classificationTemplate(classificationTypes classificationType);
    
    classificationTemplate(const std::vector<trainingExampleTemplate<T> > &trainingSet);
    classificationTemplate(const int &numInputs, const int &numOutputs);
    
    ~classificationTemplate() {}
    
    bool train(const std::vector<trainingExampleTemplate<T> > &trainingSet);
    
    std::vector<int> getK();
    void setK(const int whichModel, const int newK);
    
    
    
private:
    classificationTypes classificationType;
};

//This is here so that the old API still works as expected.
using classification = classificationTemplate<double>;
using classificationFloat = classificationTemplate<float>;

#endif