ml_pipeline.cpp

Go to the documentation of this file.

//=============================================================================================================
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "ml_pipeline.h"
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QDebug>
 
//=============================================================================================================
// STL INCLUDES
//=============================================================================================================
 
#include <stdexcept>
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace MLLIB;
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================
 
MlPipeline::MlPipeline()
{
}
 
//=============================================================================================================
 
void MlPipeline::setScaler(const MlScaler& scaler)
{
    m_scaler    = scaler;
    m_hasScaler = true;
}
 
//=============================================================================================================
 
void MlPipeline::setModel(MlModel::SPtr model)
{
    m_model = model;
}
 
//=============================================================================================================
 
void MlPipeline::fitScaler(const MlTensor& X)
{
    if (!m_hasScaler) {
        qWarning() << "MlPipeline::fitScaler – no scaler has been set.";
        return;
    }
 
    m_scaler.fit(X);
}
 
//=============================================================================================================
 
MlTensor MlPipeline::predict(const MlTensor& X) const
{
    if (!m_model) {
        throw std::runtime_error("MlPipeline::predict – no model has been set.");
    }
 
    if (m_hasScaler) {
        MlTensor Xs = m_scaler.transform(X);
        return m_model->predict(Xs);
    }
 
    return m_model->predict(X);
}