AbstractEStep.hpp

#ifndef _LDAPLUSPLUS_EM_ABSTRACTESTEP_HPP_
#define _LDAPLUSPLUS_EM_ABSTRACTESTEP_HPP_

#include <random>

#include "ldaplusplus/em/EStepInterface.hpp"
#include "ldaplusplus/utils.hpp"

namespace ldaplusplus {
namespace em {

template <typename Scalar>
class AbstractEStep : public EStepInterface<Scalar>
{
    typedef math_utils::ThreadSafePRNG<std::default_random_engine> PRNG;

    public:
        AbstractEStep(int random_state);

        virtual void e_step() override {}

    protected:
        bool converged(
            const Eigen::Matrix<Scalar, Eigen::Dynamic, 1> & gamma_old,
            const Eigen::Matrix<Scalar, Eigen::Dynamic, 1> & gamma,
            Scalar tolerance
        );

        PRNG &get_prng() { return random_; }

    private:
        // A random number generator to be used for every random number needed
        // in this E step (initialized with random_state constructor parameter)
        PRNG random_;
};


}  // namespace em
}  // namespace ldaplusplus

#endif  // _LDAPLUSPLUS_EM_ABSTRACTESTEP_HPP_