nn_alg.hh

Go to the documentation of this file.

#ifndef _NN_ALGHH_
#define _NN_ALGHH_

#include <vector>

#include <openmodeller/om.hh>

#include "nn.h"

/**************************************************/
/****************** Neural Network ****************/

/* 
 * Algorithm to generate models using Neural Network.
 */
class NNAlgorithm : public AlgorithmImpl
{

public:

  NNAlgorithm();

  ~NNAlgorithm();

  int needNormalization();

  int initialize();

  int iterate();

  float getProgress() const;

  int done() const;


  //
  // Methods used to project the model
  //
    
  Scalar getValue( const Sample& x ) const;
  

 
protected:

  virtual void _getConfiguration( ConfigurationPtr& ) const;

  virtual void _setConfiguration( const ConstConfigurationPtr& );


  bool _done;

  int _num_layers;

  int num_presences;

  int num_absences;


  // Algorithm's variables
  
  Network network;

  vector<vector<double> > vector_input; // [num_patterns][_num_layers]

  vector<vector<double> > vector_output;// [num_pattern][_nn_parameter.outp]

  nn_parameter _nn_parameter;

  double *outputs;

  unsigned long amount_epoch; // Number of epoch in training

  float _progress; // Percent of training

  int converged; // Used to verify if the training by minimum error converged


  // Work with occurrences 

  OccurrencesPtr absences;

  OccurrencesPtr presences;
};


#endif