dismec.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_DISMEC_H
#define DISMEC_DISMEC_H
 
 
#include "spec.h"
#include "parallel/numa.h"
#include "utils/hyperparams.h"
 
namespace dismec
{
    class DiSMECTraining : public TrainingSpec {
    public:
        DiSMECTraining(std::shared_ptr<const DatasetBase> data, HyperParameters hyper_params,
                       std::shared_ptr<WeightingScheme> weighting,
                       std::shared_ptr<init::WeightInitializationStrategy> init,
                       std::shared_ptr<postproc::PostProcessFactory> post_proc,
                       std::shared_ptr<TrainingStatsGatherer> gatherer,
                       bool use_sparse,
                       RegularizerSpec regularizer, LossType loss);
 
        [[nodiscard]] std::shared_ptr<objective::Objective> make_objective() const override;
        [[nodiscard]] std::unique_ptr<solvers::Minimizer> make_minimizer() const override;
        [[nodiscard]] std::unique_ptr<init::WeightsInitializer> make_initializer() const override;
        [[nodiscard]] std::shared_ptr<model::Model> make_model(long num_features, model::PartialModelSpec spec) const override;
 
        void update_minimizer(solvers::Minimizer& base_minimizer, label_id_t label_id) const override;
        void update_objective(objective::Objective& base_objective, label_id_t label_id) const override;
 
        [[nodiscard]] std::unique_ptr<postproc::PostProcessor> make_post_processor(const std::shared_ptr<objective::Objective>& objective) const override;
 
        TrainingStatsGatherer& get_statistics_gatherer() override;
    private:
        HyperParameters m_NewtonSettings;
        std::shared_ptr<WeightingScheme> m_Weighting;
        bool m_UseSparseModel = false;
 
        // initial conditions
        std::shared_ptr<init::WeightInitializationStrategy> m_InitStrategy;
 
        // post processing
        std::shared_ptr<postproc::PostProcessFactory> m_PostProcessor;
 
        parallel::NUMAReplicator<const GenericFeatureMatrix> m_FeatureReplicator;
 
        std::shared_ptr<TrainingStatsGatherer> m_StatsGather;
 
        double m_BaseEpsilon;
        RegularizerSpec m_Regularizer;
        LossType m_Loss;
    };
}
 
#endif //DISMEC_DISMEC_H