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