dense.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_DENSE_H
#define DISMEC_DENSE_H
 
#include "model/model.h"
 
namespace dismec::model {
 
    class DenseModel : public Model
    {
    public:
        using WeightMatrix = types::DenseColMajor<real_t>;
        using weight_matrix_ptr = std::shared_ptr<WeightMatrix>;
 
        explicit DenseModel(const weight_matrix_ptr& weights);
 
        DenseModel(weight_matrix_ptr weights, PartialModelSpec partial);
 
        DenseModel(long num_features, long num_labels);
 
        DenseModel(long num_features, PartialModelSpec partial);
 
        [[nodiscard]] bool has_sparse_weights() const final { return false; }
 
        [[nodiscard]] long num_features() const override;
 
        [[nodiscard]] const WeightMatrix& get_raw_weights() const { return *m_Weights; }
 
    private:
        void get_weights_for_label_unchecked(label_id_t label, Eigen::Ref<DenseRealVector> target) const override;
 
        void set_weights_for_label_unchecked(label_id_t label, const WeightVectorIn& weights) override;
 
        void predict_scores_unchecked(const FeatureMatrixIn& instances,
                                      PredictionMatrixOut target) const override;
 
        weight_matrix_ptr m_Weights;
    };
 
}
 
#endif //DISMEC_DENSE_H