ml_pipeline.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef ML_PIPELINE_H
#define ML_PIPELINE_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "ml_global.h"
#include "ml_model.h"
#include "ml_scaler.h"
 
//=============================================================================================================
// DEFINE NAMESPACE MLLIB
//=============================================================================================================
 
namespace MLLIB{
 
//=============================================================================================================
class MLSHARED_EXPORT MlPipeline
{
public:
    //=========================================================================================================
    MlPipeline();
 
    //=========================================================================================================
    void setScaler(const MlScaler& scaler);
 
    //=========================================================================================================
    void setModel(MlModel::SPtr model);
 
    //=========================================================================================================
    void fitScaler(const MlTensor& X);
 
    //=========================================================================================================
    MlTensor predict(const MlTensor& X) const;
 
private:
    MlScaler        m_scaler;           
    MlModel::SPtr   m_model;            
    bool            m_hasScaler = false; 
};
 
} // namespace MLLIB
 
#endif // ML_PIPELINE_H