standard_scaler.h

Go to the documentation of this file.

 
#if !defined(ULTRA_STANDARD_SCALER_H)
#define      ULTRA_STANDARD_SCALER_H
 
#include "kernel/gp/src/dataframe.h"
 
#include <string>
#include <utility>
#include <vector>
 
namespace ultra::src
{

class standard_scaler
{
public:
  struct column_stats
  {
    double anchor {0.0};   
    double scale {1.0};    
    double mean {0.0};     
    double std_dev {1.0};  
  };
 
  struct options
  {
    bool scale_inputs {true};   
    bool scale_output {false};  
  };
 
  standard_scaler() = default;
  explicit standard_scaler(options) noexcept;
 
  void fit(const dataframe &);
  void transform(dataframe &) const;
 
  [[nodiscard]] double inverse_transform_target(double) const;
 
  [[nodiscard]] bool is_fitted() const noexcept { return is_fitted_; }
  [[nodiscard]] bool is_valid() const noexcept;
 
private:
  std::vector<std::pair<std::string, domain_t>> schema_ {};
  std::vector<column_stats> input_stats_ {};
  column_stats output_stats_ {};
 
  options opt_ {};
  bool is_fitted_ {false};
};
 
}  // namespace ultra::src
 
#endif  // include guard