SemiSupervisedMStep.hpp

#ifndef _LDAPLUSPLUS_EM_SEMISUPERVISEDMSTEP_HPP_
#define _LDAPLUSPLUS_EM_SEMISUPERVISEDMSTEP_HPP_


#include "ldaplusplus/em/FastSupervisedMStep.hpp"

namespace ldaplusplus {
namespace em {


template <typename Scalar>
class SemiSupervisedMStep : public FastSupervisedMStep<Scalar>
{
    public:
        SemiSupervisedMStep(
            size_t m_step_iterations = 10,
            Scalar m_step_tolerance = 1e-2,
            Scalar regularization_penalty = 1e-2
        ) : FastSupervisedMStep<Scalar>(
                m_step_iterations,
                m_step_tolerance,
                regularization_penalty
            )
        {}

        virtual void doc_m_step(
            const std::shared_ptr<corpus::Document> doc,
            const std::shared_ptr<parameters::Parameters> v_parameters,
            std::shared_ptr<parameters::Parameters> m_parameters
        ) override;
};

}  // namespace em
}  // namespace ldaplusplus

#endif  // _LDAPLUSPLUS_EM_SEMISUPERVISEDMSTEP_HPP_