SpectralCluster.h 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. /**
  2. * @file SpectralCluster.h
  3. * @brief spectral clustering by kmeans-clustering of eigenvectors
  4. * @author Erik Rodner
  5. * @date 11/13/2007
  6. */
  7. #ifndef SPECTRALCLUSTERINCLUDE
  8. #define SPECTRALCLUSTERINCLUDE
  9. #include "core/vector/VectorT.h"
  10. #include "core/vector/MatrixT.h"
  11. #include "ClusterAlgorithm.h"
  12. #include "KMeans.h"
  13. namespace OBJREC {
  14. /** spectral clustering by kmeans-clustering of eigenvectors */
  15. class SpectralCluster : public ClusterAlgorithm
  16. {
  17. protected:
  18. int noClasses;
  19. double alpha;
  20. KMeans kmeans;
  21. enum {
  22. L_UNNORMALIZED = 0,
  23. L_RW_NORMALIZED
  24. };
  25. virtual void computeLaplacian ( const NICE::VVector & features,
  26. // refactor-nice.pl: check this substitution
  27. // old: Matrix & laplacian,
  28. NICE::Matrix & laplacian,
  29. int method, double alpha );
  30. virtual void getSimilarityMatrix ( const NICE::VVector & features,
  31. // refactor-nice.pl: check this substitution
  32. // old: Matrix & laplacian, double alpha );
  33. NICE::Matrix & laplacian, double alpha );
  34. public:
  35. /** simple constructor */
  36. SpectralCluster (int _noClasses, double alpha);
  37. /** simple destructor */
  38. virtual ~SpectralCluster();
  39. void cluster ( const NICE::VVector & features,
  40. NICE::VVector & prototypes,
  41. std::vector<double> & weights,
  42. std::vector<int> & assignment );
  43. };
  44. } // namespace
  45. #endif