extended_infomax.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef EXTENDED_INFOMAX_H
#define EXTENDED_INFOMAX_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "dsp_global.h"
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
#include <Eigen/Dense>
 
//=============================================================================================================
// DEFINE NAMESPACE UTILSLIB
//=============================================================================================================
 
namespace UTILSLIB {
 
//=============================================================================================================
struct DSPSHARED_EXPORT InfomaxResult {
    Eigen::MatrixXd matUnmixing;    
    Eigen::MatrixXd matMixing;      
    Eigen::MatrixXd matSources;     
    int nIterations;                
    bool converged;                 
};
 
//=============================================================================================================
class DSPSHARED_EXPORT ExtendedInfomax {
public:
    static InfomaxResult compute(
        const Eigen::MatrixXd& matData,
        int nComponents = -1,
        int maxIterations = 200,
        double learningRate = 0.001,
        double tolerance = 1e-7,
        bool extendedMode = true,
        unsigned int seed = 0);
 
private:
    static Eigen::VectorXd estimateSignVector(const Eigen::MatrixXd& matSources);
};
 
} // namespace UTILSLIB
 
#endif // EXTENDED_INFOMAX_H