classification.h

Go to the documentation of this file.

 
#ifndef CLASSIFICATION_H
#define CLASSIFICATION_H
 
#include <vector>
#include "modelSet.h"
 
template<typename T>
class classificationTemplate final : 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) override;
    
    std::vector<int> getK();
 
    void setK(const int whichModel, const int newK);
     
private:
    classificationTypes classificationType;
};
 
namespace rapidLib
{
    //This is here so that the old API still works as expected.
    using classification = classificationTemplate<double>;
    using classificationFloat = classificationTemplate<float>;
}
 
#endif