np_neal_algorithm2.h

Go to the documentation of this file.

#pragma once

#include <random>

#include <statistics/distribution.h>
#include <membertrix.h>
#include <np_update_cluster_population.h>
#include <statistics/dirichlet.h>
#include <np_statistics.h>

class NealAlgorithm2: public UpdateClusterPopulation {
    private:
        std::default_random_engine _generator;

        distribution_t & _likelihood;
    
        dirichlet_process & _nonparametrics;
        
        double _alpha;

        // verbosity
        char _verbosity;

        // statistics
        statistics_t _statistics;

        /*
        distribution_t _likelihood;
        
        distribution_t _nonparametrics;
        
        distribution_t _prior;
        */

    public:
        NealAlgorithm2(
            random_engine_t & generator,
            distribution_t & likelihood,
            dirichlet_process & nonparametrics
        );

        void update(
            membertrix & cluster_matrix,
            const data_ids_t & data_ids
        );

        void printStatistics();
};