EStepInterface.hpp

#ifndef _LDAPLUSPLUS_EM_ESTEPINTERFACE_HPP_
#define _LDAPLUSPLUS_EM_ESTEPINTERFACE_HPP_

#include <Eigen/Core>

#include "ldaplusplus/Document.hpp"
#include "ldaplusplus/events/Events.hpp"
#include "ldaplusplus/Parameters.hpp"

namespace ldaplusplus {
namespace em {


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

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

        virtual void e_step()=0;

        virtual ~EStepInterface(){};
};

}  // namespace em
}  // namespace ldaplusplus

#endif // _LDAPLUSPLUS_EM_ESTEPINTERFACE_HPP_