SemiSupervisedEStep.hpp

#ifndef _LDAPLUSPLUS_EM_SEMISUPERVISEDESTEP_HPP_
#define _LDAPLUSPLUS_EM_SEMISUPERVISEDESTEP_HPP_


#include <memory>

#include "ldaplusplus/em/EStepInterface.hpp"

namespace ldaplusplus {
namespace em {


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

    public:
        SemiSupervisedEStep(
            std::shared_ptr<EStepInterface<Scalar> > supervised_step,
            std::shared_ptr<EStepInterface<Scalar> > unsupervised_step
        );
        virtual ~SemiSupervisedEStep();

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

        void e_step() override;

    private:
        std::shared_ptr<EStepInterface<Scalar> > supervised_step_;
        std::shared_ptr<EStepInterface<Scalar> > unsupervised_step_;

        std::shared_ptr<events::EventListenerInterface> event_forwarder_;
};

}  // namespace em
}  // namespace ldaplusplus

#endif  // _LDAPLUSPLUS_EM_SEMISUPERVISEDESTEP_HPP_