Simple scaler → model pipeline. More...

Public Member Functions
	MlPipeline ()
void	setScaler (const MlScaler &scaler)
void	setModel (MlModel::SPtr model)
void	fitScaler (const MlTensor &X)
MlTensor	predict (const MlTensor &X) const

Detailed Description

Simple scaler → model pipeline.

Definition at line 56 of file ml_pipeline.h.

Constructor & Destructor Documentation

MlPipeline::MlPipeline ( )

Default constructor.

Definition at line 63 of file ml_pipeline.cpp.

void MlPipeline::fitScaler ( const MlTensor & X )

Fit the scaler on the given features (does not train the model).

Parameters

[in] X Feature matrix.

Definition at line 84 of file ml_pipeline.cpp.

MlTensor MlPipeline::predict ( const MlTensor & X ) const

Scale (if scaler set) and predict.

Parameters

[in] X Feature matrix.

Definition at line 96 of file ml_pipeline.cpp.

void MlPipeline::setModel ( MlModel::SPtr model )

Set the model.

Parameters

[in] model Shared pointer to the model.

Definition at line 77 of file ml_pipeline.cpp.

void MlPipeline::setScaler ( const MlScaler & scaler )

Set the feature scaler.

Parameters

[in] scaler The scaler to use.

Definition at line 69 of file ml_pipeline.cpp.

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