SpectralCluster.h 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  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. #ifdef NOVISUAL
  10. #include <vislearning/nice_nonvis.h>
  11. #else
  12. #include <vislearning/nice.h>
  13. #endif
  14. #include "ClusterAlgorithm.h"
  15. #include "KMeans.h"
  16. namespace OBJREC {
  17. /** spectral clustering by kmeans-clustering of eigenvectors */
  18. class SpectralCluster : public ClusterAlgorithm
  19. {
  20. protected:
  21. int noClasses;
  22. double alpha;
  23. KMeans kmeans;
  24. enum {
  25. L_UNNORMALIZED = 0,
  26. L_RW_NORMALIZED
  27. };
  28. virtual void computeLaplacian ( const NICE::VVector & features,
  29. // refactor-nice.pl: check this substitution
  30. // old: Matrix & laplacian,
  31. NICE::Matrix & laplacian,
  32. int method, double alpha );
  33. virtual void getSimilarityMatrix ( const NICE::VVector & features,
  34. // refactor-nice.pl: check this substitution
  35. // old: Matrix & laplacian, double alpha );
  36. NICE::Matrix & laplacian, double alpha );
  37. public:
  38. /** simple constructor */
  39. SpectralCluster (int _noClasses, double alpha);
  40. /** simple destructor */
  41. virtual ~SpectralCluster();
  42. void cluster ( const NICE::VVector & features,
  43. NICE::VVector & prototypes,
  44. std::vector<double> & weights,
  45. std::vector<int> & assignment );
  46. };
  47. } // namespace
  48. #endif