dict_embd_vectorizer.h

Go to the documentation of this file.

// Copyright 2021 CEA LIST
// SPDX-FileCopyrightText: 2022 CEA LIST <gael.de-chalendar@cea.fr>
//
// SPDX-License-Identifier: MIT
 
#ifndef DEEPLIMA_SRC_INFERENCE_EIGEN_DICT_EMBD_VECTORIZER_H
#define DEEPLIMA_SRC_INFERENCE_EIGEN_DICT_EMBD_VECTORIZER_H
 
#include <vector>
#include <cassert>
 
namespace deeplima
{
 
template <class DH, class D, class Matrix>
class DictEmbdVectorizer : public Matrix
{
public:
  DictEmbdVectorizer() {}
 
  // must be called before init
  void set_dicts(const DH& dicts)
  {
    m_dicts = dicts;
  }
 
  void init(const DH& dicts, int64_t max_time, uint64_t max_feat)
  {
    assert(dicts.size() == max_feat);
    m_dicts = dicts;
    m_pos.reserve(max_feat);
    Eigen::Index dim = 0;
    for ( const auto& d : m_dicts )
    {
      m_pos.push_back(dim);
      dim += d.dim();
    }
    Matrix::init(max_time, dim);
  }
 
  void init(int64_t max_time, uint64_t max_feat)
  {
    assert(m_dicts.size() > 0);
    assert(m_dicts.size() == max_feat);
    m_pos.reserve(max_feat);
    Eigen::Index dim = 0;
    for ( const auto& d : m_dicts )
    {
      m_pos.push_back(dim);
      dim += d.dim();
    }
    Matrix::init(max_time, dim);
  }
 
  inline void set(uint64_t time, uint64_t feat, typename D::value_t value)
  {
    assert(m_dicts.size() == m_pos.size());
    assert(feat < m_dicts.size());
    m_dicts[feat].get(value, Matrix::get_tensor(), time, m_pos[feat]);
  }
 
protected:
  DH m_dicts;
  std::vector<Eigen::Index> m_pos;
};
 
} // namespace deeplima
 
#endif