np_jain_neal_algorithm.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 JainNealAlgorithm: public UpdateClusterPopulation {
    private:
        std::default_random_engine _generator;

        distribution_t & _likelihood;

        split_method_t _split_method;

        dirichlet_process & _nonparametrics;

        membertrix * _cluster_matrix;

        double _alpha;

        // verbosity
        char _verbosity;

        // statistics
        statistics_t _statistics;

        double ratioStateProb(bool split, int nc0, int nc1, int nc);

        double ratioProposal(bool split, int N, int C = 2);

        void checkLikelihoods(double l1, double l2, step_t statistics_step, double &l21, bool &accept, bool &overwrite);

        void propose_split(data_ids_t & move, data_ids_t & remain, data_id_t data_i, data_id_t data_j, 
                cluster_id_t current_cluster_id, cluster_t & new_cluster, split_method_t split_method);

        // split a single cluster
        bool split(data_ids_t data_ids, cluster_id_t current_cluster);

        // merge the given clusters
        bool merge(data_ids_t data_ids, std::vector<cluster_id_t> & current_clusters);
    public:
        JainNealAlgorithm(
            random_engine_t & generator,
            distribution_t & likelihood,
            dirichlet_process & nonparametrics
        );

        ~JainNealAlgorithm();

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

        void printStatistics();
};