QuantileNormalizer2.h

#ifndef _theplu_yat_normalizer_quantile_normalizer2_
#define _theplu_yat_normalizer_quantile_normalizer2_

// $Id: QuantileNormalizer2.h 4191 2022-07-18 10:14:18Z peter $

/*
  Copyright (C) 2022 Peter Johansson

  This file is part of the yat library, https://dev.thep.lu.se/yat

  The yat library is free software; you can redistribute it and/or
  modify it under the terms of the GNU General Public License as
  published by the Free Software Foundation; either version 3 of the
  License, or (at your option) any later version.

  The yat library is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with yat. If not, see <https://www.gnu.org/licenses/>.
*/

#include "yat/regression/LinearInterpolation.h"
#include "yat/utility/Vector.h"

namespace theplu {
namespace yat {
  namespace utility {
    class MatrixBase;
    class MatrixMutable;
  }
namespace normalizer {

  class QuantileNormalizer2
  {
  public:
    QuantileNormalizer2(void);

    QuantileNormalizer2(const QuantileNormalizer2& other);

    QuantileNormalizer2(QuantileNormalizer2&& other);

    QuantileNormalizer2& operator=(const QuantileNormalizer2& rhs);

    QuantileNormalizer2& operator=(QuantileNormalizer2&& rhs);

    void train(const utility::MatrixBase& data);

    void operator()(const utility::MatrixBase& input,
                    utility::MatrixMutable& result);

    void normalize(const utility::MatrixBase& input,
                   utility::MatrixMutable& result) const;

    void normalize(const utility::VectorBase& input,
                   utility::VectorMutable& result) const;

  private:
    friend void swap(QuantileNormalizer2& lhs, QuantileNormalizer2& rhs);
    utility::Vector distribution_;
    mutable std::unique_ptr<regression::LinearInterpolation> interpolator_;

    void create_index(const utility::MatrixBase& X,
                      std::vector<std::vector<size_t>>& index) const;

    regression::LinearInterpolation& interpolator(void) const;

    void normalize(utility::VectorMutable& result,
                   const std::vector<size_t>& index) const;
    void normalize(utility::VectorMutable& result,
                   const std::vector<size_t>& index,
                   regression::LinearInterpolation& interpolator) const;

    void normalize(utility::MatrixMutable& result,
                   const std::vector<std::vector<size_t>>& index) const;
    void normalize(utility::MatrixMutable& result,
                   const std::vector<std::vector<size_t>>& index,
                   regression::LinearInterpolation& interpolator) const;

    void train(const utility::MatrixBase& X,
               const std::vector<std::vector<size_t>>& index);
  };

  void swap(QuantileNormalizer2& lhs, QuantileNormalizer2& rhs);

}}} // end of namespace normalizer, yat and thep
#endif