Parameters.hpp

#ifndef _LDAPLUSPLUS_PARAMETERS_HPP_
#define _LDAPLUSPLUS_PARAMETERS_HPP_


#include <utility>

#include <Eigen/Core>

namespace ldaplusplus {
namespace parameters {


struct Parameters
{
};


template <typename Scalar = double>
struct ModelParameters : public Parameters
{
    ModelParameters() {}
    ModelParameters(
        Eigen::Matrix<Scalar, Eigen::Dynamic, 1> a,
        Eigen::Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic> b
    ) : alpha(std::move(a)),
        beta(std::move(b))
    {}

    Eigen::Matrix<Scalar, Eigen::Dynamic, 1> alpha;
    Eigen::Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic> beta;
};


template <typename Scalar = double>
struct SupervisedModelParameters : public ModelParameters<Scalar>
{
    SupervisedModelParameters() {}
    SupervisedModelParameters(
        Eigen::Matrix<Scalar, Eigen::Dynamic, 1> a,
        Eigen::Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic> b,
        Eigen::Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic> e
    ) : ModelParameters<Scalar>(a, b),
        eta(std::move(e))
    {}

    Eigen::Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic> eta;
};


template <typename Scalar = double>
struct VariationalParameters : public Parameters
{
    VariationalParameters() {}
    VariationalParameters(
        Eigen::Matrix<Scalar, Eigen::Dynamic, 1> g,
        Eigen::Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic> p
    ) : gamma(std::move(g)),
        phi(std::move(p))
    {}

    Eigen::Matrix<Scalar, Eigen::Dynamic, 1> gamma;
    Eigen::Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic> phi;
};


template <typename Scalar = double>
struct SupervisedCorrespondenceVariationalParameters : public VariationalParameters<Scalar>
{
    SupervisedCorrespondenceVariationalParameters() {}
    SupervisedCorrespondenceVariationalParameters(
        Eigen::Matrix<Scalar, Eigen::Dynamic, 1> g,
        Eigen::Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic> p,
        Eigen::Matrix<Scalar, Eigen::Dynamic, 1> t
    ) : VariationalParameters<Scalar>(g, p),
        tau(std::move(t))
    {}

    Eigen::Matrix<Scalar, Eigen::Dynamic, 1> tau;
};

}  // namespace parameters
}  // namespace ldaplusplus

#endif  // _LDAPLUSPLUS_PARAMETERS_HPP_