e_step_utils.hpp

#ifndef _LDAPLUSPLUS_ESTEPUTILS_HPP_
#define _LDAPLUSPLUS_ESTEPUTILS_HPP_

#include <Eigen/Core>

namespace ldaplusplus {
namespace e_step_utils {

    template <typename Scalar>
    using MatrixX = Eigen::Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic>;
    template <typename Scalar>
    using VectorX = Eigen::Matrix<Scalar, Eigen::Dynamic, 1>;
    using Eigen::VectorXi;
    using Eigen::Ref;

    template <typename Scalar>
    Scalar compute_unsupervised_likelihood(
        const VectorXi & X,
        const VectorX<Scalar> &alpha,
        const MatrixX<Scalar> &beta,
        const MatrixX<Scalar> &phi,
        const VectorX<Scalar> &gamma
    );

    template <typename Scalar>
    Scalar compute_supervised_likelihood(
        const VectorXi & X,
        int y,
        const VectorX<Scalar> &alpha,
        const MatrixX<Scalar> &beta,
        const MatrixX<Scalar> &eta,
        const MatrixX<Scalar> &phi,
        const VectorX<Scalar> &gamma
    );
    template <typename Scalar>
    Scalar compute_supervised_likelihood(
        const VectorXi & X,
        int y,
        const VectorX<Scalar> &alpha,
        const MatrixX<Scalar> &beta,
        const MatrixX<Scalar> &eta,
        const MatrixX<Scalar> &phi,
        const VectorX<Scalar> &gamma,
        const VectorX<Scalar> &h
    );

    template <typename Scalar>
    Scalar compute_supervised_multinomial_likelihood(
        const VectorXi & X,
        int y,
        const VectorX<Scalar> &alpha,
        const MatrixX<Scalar> &beta,
        const MatrixX<Scalar> &eta,
        const MatrixX<Scalar> &phi,
        const VectorX<Scalar> &gamma,
        Scalar prior_y,
        Scalar mu,
        Scalar portion
    );

    template <typename Scalar>
    Scalar compute_supervised_correspondence_likelihood(
        const VectorXi & X,
        int y,
        const VectorX<Scalar> &alpha,
        const MatrixX<Scalar> &beta,
        const MatrixX<Scalar> &eta,
        const MatrixX<Scalar> &phi,
        const VectorX<Scalar> &gamma,
        const VectorX<Scalar> &tau,
        Scalar mu,
        Scalar portion
    );

    template <typename Scalar>
    void compute_h(
        const VectorXi & X,
        const VectorX<Scalar> & X_ratio,
        const MatrixX<Scalar> &eta,
        const MatrixX<Scalar> &phi,
        Ref<VectorX<Scalar> > h
    );

    template <typename Scalar>
    void compute_supervised_phi_gamma(
        const VectorXi & X,
        const VectorX<Scalar> & X_ratio,
        int y,
        const MatrixX<Scalar> & beta,
        const MatrixX<Scalar> & eta,
        size_t fixed_point_iterations,
        Ref<MatrixX<Scalar> > phi,
        Ref<VectorX<Scalar> > gamma,
        Ref<VectorX<Scalar> > h
    );

    template <typename Scalar>
    void compute_gamma(
        const VectorXi & X,
        const VectorX<Scalar> & alpha,
        const MatrixX<Scalar> & phi,
        Ref<VectorX<Scalar> > gamma
    );

    template <typename Scalar>
    void compute_unsupervised_phi(
        const MatrixX<Scalar> & beta,
        const VectorX<Scalar> & gamma,
        Ref<MatrixX<Scalar> > phi
    );

    template <typename Scalar>
    void compute_supervised_approximate_phi(
        const VectorX<Scalar> & X_ratio,
        int num_words,
        int y,
        const MatrixX<Scalar> & beta,
        const MatrixX<Scalar> & eta,
        const VectorX<Scalar> & gamma,
        Scalar C,
        Ref<MatrixX<Scalar> > phi
    );

    template <typename Scalar>
    void compute_supervised_multinomial_phi(
        const VectorXi & X,
        int y,
        const MatrixX<Scalar> & beta,
        const MatrixX<Scalar> & eta,
        const VectorX<Scalar> & gamma,
        Scalar eta_weight,
        Ref<MatrixX<Scalar> > phi
    );

    template <typename Scalar>
    void compute_supervised_correspondence_phi(
        const VectorXi & X,
        int y,
        const MatrixX<Scalar> & beta,
        const MatrixX<Scalar> & eta,
        const VectorX<Scalar> & gamma,
        const VectorX<Scalar> & tau,
        Ref<MatrixX<Scalar> > phi
    );

    template <typename Scalar>
    void compute_supervised_correspondence_tau(
        const VectorXi & X,
        int y,
        const MatrixX<Scalar> & eta,
        const MatrixX<Scalar> & phi,
        Ref<VectorX<Scalar> > tau
    );
} // namespace e_step_utils

} // namespace ldaplusplus
#endif  // _LDAPLUSPLUS_ESTEPUTILS_HPP_