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 final : 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) override;
  
  float getTrainingProgress();
  
  std::vector<size_t> getNumEpochs() const;
  
  void setNumEpochs(const size_t &epochs);
  
  std::vector<size_t> getNumHiddenLayers() const;
  
  void setNumHiddenLayers(const int &num_hidden_layers);
  
  std::vector<size_t> getNumHiddenNodes() const;
  
  void setNumHiddenNodes(const int &num_hidden_nodes);
  
private:
  size_t numEpochs { 500 }; //Temporary -- also should be part of nn only. -mz
  size_t numHiddenLayers { 1 }; //Temporary -- this should be part of the nn class. -mz
  size_t numHiddenNodes; //Temporary -- also should be part of nn only. -mz
  bool created;
};
 
namespace rapidLib
{
//This is here so the old API still works
using regression = regressionTemplate<double>;
using regressionFloat = regressionTemplate<float>;
};
 
#endif