np_mcmc.h

Go to the documentation of this file.

#pragma once

#include <vector>
#include <random>

#include <membertrix.h>

#include <np_data.h>

#include <np_init_clusters.h>
#include <np_update_clusters.h>
#include <np_update_cluster_population.h>

class MCMC {
    private:
        std::default_random_engine & _generator;
    
        membertrix _membertrix;
        
        InitClusters & _init_clusters;

        UpdateClusters & _update_clusters;

        UpdateClusterPopulation & _update_cluster_population;

        char _verbosity;

        bool _random_scan;

        int _subset_count;

        membertrix _max_likelihood_membertrix;

        double _max_likelihood;

        distribution_t & _likelihood;
    protected:
        void considerMaxLikelihood();

    public:
        MCMC(
                std::default_random_engine & generator,
                InitClusters & init_clusters, 
                UpdateClusters & update_clusters, 
                UpdateClusterPopulation & update_cluster_population,
                int subset_count,
                distribution_t & likelihood
            );

        void run(dataset_t & dataset, int T);

        const membertrix & getMembershipMatrix() const;

        const membertrix & getMaxLikelihoodMatrix() const;
};