SupervisedEStep.hpp

#ifndef _LDAPLUSPLUS_EM_SUPERVISEDESTEP_HPP_
#define _LDAPLUSPLUS_EM_SUPERVISEDESTEP_HPP_

#include "ldaplusplus/em/AbstractEStep.hpp"

namespace ldaplusplus {
namespace em {


template <typename Scalar>
class SupervisedEStep : public AbstractEStep<Scalar>
{
    typedef Eigen::Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic> MatrixX;
    typedef Eigen::Matrix<Scalar, Eigen::Dynamic, 1> VectorX;

    public:
        SupervisedEStep(
            size_t e_step_iterations = 10,
            Scalar e_step_tolerance = 1e-2,
            size_t fixed_point_iterations = 20,
            Scalar compute_likelihood = 1.0,
            int random_state = 0
        );

        std::shared_ptr<parameters::Parameters> doc_e_step(
            const std::shared_ptr<corpus::Document> doc,
            const std::shared_ptr<parameters::Parameters> parameters
        ) override;

    private:
        // The maximum number of iterations in E-step.
        size_t e_step_iterations_;
        // The maximum number of iterations while maximizing phi in E-step.
        size_t fixed_point_iterations_;
        // The convergence tolerance for the maximazation of the ELBO w.r.t.
        // phi and gamma in E-step.
        Scalar e_step_tolerance_;
        // Compute the likelihood of that many documents (pecentile)
        Scalar compute_likelihood_;
};

}  // namespace em
}  // namespace ldaplusplus

#endif  // _LDAPLUSPLUS_EM_SUPERVISEDESTEP_HPP_