mlpack
Public Member Functions | List of all members
mlpack::data::ZCAWhitening Class Reference

A simple ZCAWhitening class. More...

#include <zca_whitening.hpp>

Public Member Functions

 ZCAWhitening (double eps=0.00005)
 A constructor to set the regularization parameter. More...
 
template<typename MatType >
void Fit (const MatType &input)
 Function to fit features, to find out the min max and scale. More...
 
template<typename MatType >
void Transform (const MatType &input, MatType &output)
 Function for ZCA whitening. More...
 
template<typename MatType >
void InverseTransform (const MatType &input, MatType &output)
 Function to retrieve original dataset. More...
 
const arma::vec & ItemMean () const
 Get the mean row vector.
 
const arma::vec & EigenValues () const
 Get the eigenvalues vector.
 
const arma::mat & EigenVectors () const
 Get the eigenvector.
 
double Epsilon () const
 Get the regularization parameter.
 
template<typename Archive >
void serialize (Archive &ar, const uint32_t)
 

Detailed Description

A simple ZCAWhitening class.

Whitens a matrix using the eigendecomposition of the covariance matrix. Whitening means the covariance matrix of the result is the identity matrix.

For whitening related formula and more info, check the link below. http://ufldl.stanford.edu/tutorial/unsupervised/PCAWhitening/

arma::mat input;
Load("train.csv", input);
arma::mat output;
// Fit the features.
scale.Fit(input)
// Scale the features.
scale.Transform(input, output);
// Retransform the input.
scale.InverseTransform(output, input);

Constructor & Destructor Documentation

◆ ZCAWhitening()

mlpack::data::ZCAWhitening::ZCAWhitening ( double  eps = 0.00005)
inline

A constructor to set the regularization parameter.

Parameters
epsRegularization parameter.

Member Function Documentation

◆ Fit()

template<typename MatType >
void mlpack::data::ZCAWhitening::Fit ( const MatType &  input)
inline

Function to fit features, to find out the min max and scale.

Parameters
inputDataset to fit.

◆ InverseTransform()

template<typename MatType >
void mlpack::data::ZCAWhitening::InverseTransform ( const MatType &  input,
MatType &  output 
)
inline

Function to retrieve original dataset.

Parameters
inputScaled dataset.
outputOutput matrix with original Dataset.

◆ Transform()

template<typename MatType >
void mlpack::data::ZCAWhitening::Transform ( const MatType &  input,
MatType &  output 
)
inline

Function for ZCA whitening.

Parameters
inputDataset to scale features.
outputOutput matrix with whitened features.

The documentation for this class was generated from the following file: