np_triadic_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 TriadicAlgorithm: public UpdateClusterPopulation {
    private:
        std::default_random_engine _generator;

        distribution_t & _likelihood;

        split_method_t _split_method;

        dirichlet_process & _nonparametrics;

        membertrix * _cluster_matrix;

        double _alpha;

        double _beta;

        // verbosity
        char _verbosity;

        // statistics
        statistics_t _statistics;

        double ratioStateProb(bool split, const std::vector<int> &more, const std::vector<int> &less);

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

        double ratioR(bool split, int S, int T);

        void propose_merge(std::vector<data_ids_t> &pdata, const data_ids_t &data_ids, 
                cluster_ids_t &cluster_ids, split_method_t split_method);

        void propose_split(std::vector<data_ids_t> &pdata, const data_ids_t &data_ids, 
                cluster_ids_t &cluster_ids, cluster_t *new_cluster, split_method_t split_method);

        // split a single cluster
        bool split(const data_ids_t & data_ids, cluster_ids_t & cluster_ids);

        // merge the given clusters
        bool merge(const data_ids_t & data_ids, cluster_ids_t & clusters_ids);
    public:
        TriadicAlgorithm(
            random_engine_t & generator,
            distribution_t & likelihood,
            dirichlet_process & nonparametrics
        );

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

        void printStatistics();
};