regression.h

Go to the documentation of this file.

#ifndef regression_h
#define regression_h

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

template<typename T>
class regressionTemplate : public modelSet<T> {
public:
    regressionTemplate();
    regressionTemplate(const std::vector<trainingExampleTemplate<T> > &trainingSet);
    regressionTemplate(const int &numInputs, const int &numOutputs);
    
    ~regressionTemplate() {};
    
    bool train(const std::vector<trainingExampleTemplate<T> > &trainingSet);
    
    void setNumEpochs(const int &epochs);
    
    std::vector<int> getNumHiddenLayers();
    
    void setNumHiddenLayers(const int &num_hidden_layers);
    
private:
    int numHiddenLayers; //Temporary -- this should be part of the nn class. -mz
    int numEpochs; //Temporary -- also should be part of nn only. -mz
};

//This is here so the old API still works
using regression = regressionTemplate<double>;
using regressionFloat = regressionTemplate<float>;

#endif