ml_scaler.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef ML_SCALER_H
#define ML_SCALER_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "ml_global.h"
#include "ml_tensor.h"
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// DEFINE NAMESPACE MLLIB
//=============================================================================================================
 
namespace MLLIB{
 
//=============================================================================================================
class MLSHARED_EXPORT MlScaler
{
public:
    enum ScalerType { StandardScaler, MinMaxScaler };
 
    //=========================================================================================================
    MlScaler(ScalerType type = StandardScaler);
 
    //=========================================================================================================
    void fit(const MlTensor& data);
 
    //=========================================================================================================
    MlTensor transform(const MlTensor& data) const;
 
    //=========================================================================================================
    MlTensor fitTransform(const MlTensor& data);
 
    //=========================================================================================================
    MlTensor inverseTransform(const MlTensor& data) const;
 
private:
    ScalerType      m_type;     
    Eigen::VectorXf m_mean;    
    Eigen::VectorXf m_std;     
    Eigen::VectorXf m_min;     
    Eigen::VectorXf m_range;   
    bool            m_fitted = false;  
};
 
} // namespace MLLIB
 
#endif // ML_SCALER_H