QFD2/QFD/algorithm/EntropyWeights.h

//
// Created by Austin on 2023/10/10.
//

#ifndef ALGORITHM__ENTROPYWEIGHTS_H_
#define ALGORITHM__ENTROPYWEIGHTS_H_

#include <QVector>

typedef QVector<QVector<double> > EntropyMat;

class EntropyWeights
{
public:
    /**
     * @param mat: 输入矩阵，一行代表一个指标, 每列对应一个样本
     * @param direction，定正向指标true,负向指标false
     */
    EntropyWeights(const EntropyMat& mat, const QVector<bool>& direction);

    void setYMin(double ymin);
    void setYMax(double ymax);

    void compute(QVector<double>& weights, QVector<double>& score);

    //若归一化到[0,1], 0会出问题
    double ymin_;
    double ymax_;
    int sample_num_; //样本数
    int index_num_;  //指标数

private:
    void normalization();
    void getMinMax(const QVector<double>& in, double& min, double& max);
    void getWeights(QVector<double>& weights, QVector<double>& score);

    EntropyMat mat_;
    QVector<bool> direction_;
};

#endif //ALGORITHM__ENTROPYWEIGHTS_H_