kohonen.h

Go to the documentation of this file.

 
 
#ifndef _KOHONEN_HH_
#define _KOHONEN_HH_
 
 
#include <vector>
#include <iostream>
 
 
class Kohonen;
 


std::ostream & operator << ( std::ostream & fich, const Kohonen & koh );
std::istream & operator >> ( std::istream & fich, Kohonen & koh );
 
 

class VectCode
{
  public:
  VectCode() { cla = 0; }
  std::vector<double>   coord;
  std::vector<double> vect;
  int           cla;
};
 

class Kohonen
{

  friend std::ostream & operator << ( std::ostream &, const Kohonen & );
  friend std::istream & operator >> ( std::istream &, Kohonen & );
 
  public:

  std::vector<double>   UserData;


  Kohonen( int ndim=1, int ncdim=2 );
  Kohonen( const Kohonen & koh );
  virtual ~Kohonen();


  Kohonen & operator = ( const Kohonen & koh );


  int WinIndex() const { return( _winIndex ); }
  double        SqDistWinner() const { return( _sqDistWinner ); }
  int NDim() const { return( _nDim ); }
  int NCDim() const { return( _nCDim ); }
  int   NVCode() const { return( _vCode.size() ); }
  VectCode* VCode( unsigned i ) const { return( _vCode[i] ); }
  VectCode VCWinner() const { return(   *_vCWinner ); }
  double        LearnRay() const { return( _learnRay ); }
  void SetLearnRay( double lr ) { _learnRay = lr; }
  double        LearnFactor() const { return( _learnFactor ); }
  void SetLearnFactor( double lf ) { _learnFactor = lf; }
  double        DensityRay() const { return( _densityRay ); }
  void SetDensityRay( double dr ) { _densityRay = dr; }
  void ChangeDims( unsigned ndim, unsigned ncdim );


  void CopyVC( const std::vector<VectCode*> &vcl );
  int AddVectCode( VectCode *vc );
  int RemoveVectCode( const VectCode *vc );
  int DelVectCode( VectCode *vc );
  void Empty();
  void CreateUniformVC( unsigned nvd, double cmin=-1., double cmax=1. );
  void CreateUniformVCFromComp( unsigned c, double *comp, unsigned nvd, 
                                double cmin, double cmax );
  void RandVC( double cmin=-1., double cmax=1. );


  static void PrintVC( const VectCode *vc );
  static void PrintVect( const std::vector<double> *vect );
  static double SqVectDist( const std::vector<double> *v1, 
                            const std::vector<double> *v2 );


  int Save( const char *nom ) const;
  int Load( const char *nom );
  int ByteSize() const;


  VectCode *FindWinner( const std::vector<double> *vec );
  //VectCode *FindWinner( void *toto );
  int Learn( const std::vector<double> *vect );
  void PrintMap() const;
  double MapDist( int i1, int i2 ) const;
  double MapDist( const VectCode *vc1, const VectCode *vc2 ) const;
  double DistFact( double d ) const;
  int Density( const std::vector<double> *vect ) const;
 
 
  protected:
  unsigned              _nDim;
  unsigned              _nCDim;
  std::vector<VectCode*>        _vCode;
  VectCode              *_vCWinner;
  unsigned              _winIndex;
  double                _sqDistWinner;
  double                _learnRay;
  double                _learnFactor;
  double                _densityRay;
 
 
  private:
 
};
 
 
#endif