learn_independence_model.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSD-2-Clause
 
#ifndef cMHN_COMMON_LEARN_INDEPENDENCE_MODEL_H
#define cMHN_COMMON_LEARN_INDEPENDENCE_MODEL_H
 
#include <map>
 
namespace cMHN
{
    template<class T, class S>
    static inline auto learnIndependenceModel(
        std::map<S, T> const &pD)
    {
        constexpr pRC::Index D = typename S::Dimension();
 
        pRC::Tensor<T, D, D> theta = pRC::zero();
 
        for(auto const &[index, value] : pD)
        {
            for(pRC::Index i = 0; i < D; ++i)
            {
                if(index[i] == 1)
                {
                    theta(i, i) += value;
                }
            }
        }
        theta =
            diagonal(log(theta / (pRC::unit<pRC::Tensor<T, D, D>>() - theta)));
        return theta;
    }
 
}
 
#endif // cMHN_COMMON_LEARN_INDEPENDENCE_MODEL_H