forked from shogun-toolbox/shogun
/
ApproxJointDiagonalizer.h
76 lines (62 loc) · 1.85 KB
/
ApproxJointDiagonalizer.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
/*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* Written (W) 2013 Kevin Hughes
*
* Thanks to Andreas Ziehe and Cedric Gouy-Pailler
*/
#ifndef APPROXJOINTDIAGONALIZER_H_
#define APPROXJOINTDIAGONALIZER_H_
#include <shogun/lib/config.h>
#ifdef HAVE_EIGEN3
#include <shogun/lib/common.h>
#include <shogun/lib/SGMatrix.h>
#include <shogun/lib/SGNDArray.h>
#include <shogun/base/SGObject.h>
#include <shogun/mathematics/Math.h>
namespace shogun
{
/** @brief Class ApproxJointDiagonalizer defines an
* Approximate Joint Diagonalizer (AJD) interface.
*
* AJD finds the matrix V that best diagonalizes
* a set \f${C^1 ... C^k}\f$ of real valued symmetric
* \f$NxN\f$ matrices - \f$V*C*V^T\f$
*/
class CApproxJointDiagonalizer : public CSGObject
{
public:
/** constructor */
CApproxJointDiagonalizer() : CSGObject()
{
};
/** destructor */
virtual ~CApproxJointDiagonalizer()
{
}
/** Computes the matrix V that best diagonalizes C
* @param C the set of matrices to be diagonalized
* @param V0 an estimate of the matrix V
* @param eps machine epsilon or desired epsilon
* @param itermax maximum number of iterations
* @return V the matrix that best diagonalizes C
*/
virtual SGMatrix<float64_t> compute(SGNDArray<float64_t> C,
SGMatrix<float64_t> V0 = SGMatrix<float64_t>(NULL,0,0,false),
double eps=CMath::MACHINE_EPSILON,
int itermax=200) = 0;
/** return the matrix V that best diagonalizes C */
SGMatrix<float64_t> get_V()
{
return m_V;
}
protected:
/** the matrix V that best diagonalizes C */
SGMatrix<float64_t> m_V;
};
}
#endif //HAVE_EIGEN3
#endif //APPROXJOINTDIAGONALIZER_H_