weighting.h

Go to the documentation of this file.

// Copyright (c) 2021, Aalto University, developed by Erik Schultheis
// All rights reserved.
//
// SPDX-License-Identifier: MIT
 
#ifndef DISMEC_WEIGHTING_H
#define DISMEC_WEIGHTING_H
 
#include "matrix_types.h"
#include "fwd.h"
 
namespace dismec
{
    class PropensityModel {
    public:
        explicit PropensityModel(const DatasetBase* data, double a=0.55, double b=1.5);
        [[nodiscard]] double get_propensity(label_id_t label_id) const;
    private:
        const DatasetBase* m_Data;
        double m_A;
        double m_B;
        double m_C;
    };
 
    class WeightingScheme {
    public:
        virtual ~WeightingScheme() = default;
        [[nodiscard]] virtual double get_positive_weight(label_id_t label_id) const = 0;
        [[nodiscard]] virtual double get_negative_weight(label_id_t label_id) const = 0;
    };
 
    class ConstantWeighting : public WeightingScheme {
    public:
        ConstantWeighting(double positive_cost, double negative_cost);
        [[nodiscard]] double get_positive_weight(label_id_t label_id) const override;
        [[nodiscard]] double get_negative_weight(label_id_t label_id) const override;
    private:
        double m_PositiveCost;  
        double m_NegativeCost;  
    };
 
    class PropensityWeighting : public WeightingScheme  {
    public:
        explicit PropensityWeighting(PropensityModel model);
        [[nodiscard]] double get_positive_weight(label_id_t label_id) const override;
        [[nodiscard]] double get_negative_weight(label_id_t label_id) const override;
    private:
        PropensityModel m_Propensity;
    };
 
    class PropensityDownWeighting : public WeightingScheme  {
    public:
        explicit PropensityDownWeighting(PropensityModel model);
        [[nodiscard]] double get_positive_weight(label_id_t label_id) const override;
        [[nodiscard]] double get_negative_weight(label_id_t label_id) const override;
    private:
        PropensityModel m_Propensity;
    };
 
    class CustomWeighting : public WeightingScheme {
    public:
        explicit CustomWeighting(DenseRealVector positive_weights, DenseRealVector negative_weights);
 
        [[nodiscard]] double get_positive_weight(label_id_t label_id) const override;
        [[nodiscard]] double get_negative_weight(label_id_t label_id) const override;
    private:
        DenseRealVector m_PositiveWeights;
        DenseRealVector m_NegativeWeights;
    };
}
 
#endif //DISMEC_WEIGHTING_H