LIMA
Libre Multilingual Analyzer — C++ API
Loading...
Searching...
No Matches
deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization Class Reference

#include <deeplima/eigen_wrp/lemmatization_eigen_inference_impl.h>

Inheritance diagram for deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization:
deeplima::eigen_impl::BiRnnInferenceBase deeplima::RnnSeq2Seq deeplima::lemmatization::impl::LemmatizationImpl

Classes

struct  workbench_t
 

Public Types

using Parent = deeplima::eigen_impl::BiRnnInferenceBase
 
- Public Types inherited from deeplima::eigen_impl::BiRnnInferenceBase
typedef Eigen::MatrixXf Matrix
 
typedef Eigen::VectorXf Vector
 
typedef float Scalar
 
typedef Eigen::MatrixXf tensor_t
 
typedef EmbdUInt64FloatHolder uint_dicts_holder_t
 
typedef EmbdStrFloatHolder str_dicts_holder_t
 

Public Member Functions

virtual ~BiRnnSeq2SeqEigenInferenceForLemmatization ()=default
 
virtual void load (const std::string &fn) override
 
virtual size_t init_new_worker (size_t input_len, bool precomputed_input=false) override
 
virtual size_t get_precomputed_dim () const
 
virtual void precompute_inputs (const Eigen::MatrixXf &inputs, Eigen::MatrixXf &outputs, int64_t input_size) override
 
virtual void predict (size_t, const Eigen::MatrixXf &, int64_t, int64_t, int64_t, int64_t, std::shared_ptr< StdMatrix< uint8_t > > &, const std::vector< std::string > &) override
 
virtual void predict (size_t worker_id, const Eigen::MatrixXf &inputs, const Eigen::MatrixXf &input_feats, int64_t input_len, int64_t output_max_len, int64_t, std::vector< uint32_t > &output, const std::vector< std::string > &)
 
const std::vector< std::vector< std::string > > & get_classes () const
 
const std::vector< std::string > & get_class_names () const
 
const morph_model::morph_model_t & get_morph_model () const
 
const std::string & get_embd_fn (size_t idx) const
 
const std::vector< size_t > & get_fixed_upos () const
 
- Public Member Functions inherited from deeplima::eigen_impl::BiRnnInferenceBase
virtual ~BiRnnInferenceBase ()=default
 
const uint_dicts_holder_t & get_input_uint_dicts () const
 
const std::vector< std::string > & get_input_uint_dicts_names () const
 
const str_dicts_holder_t & get_input_str_dicts () const
 
const std::vector< std::string > & get_input_str_dicts_names () const
 
const std::vector< std::vector< std::string > > & get_output_str_dicts () const
 
const std::vector< std::string > & get_output_str_dicts_names () const
 

Protected Member Functions

virtual void convert_from_torch (const std::string &fn) override
 
- Protected Member Functions inherited from deeplima::eigen_impl::BiRnnInferenceBase
virtual void convert_dicts_and_embeddings (const nets::BiRnnClassifierImpl &src)
 

Protected Attributes

std::vector< std::string > m_class_names
 
std::vector< std::vector< std::string > > m_classes
 
std::vector< std::string > m_embd_fn
 
morph_model::morph_model_t m_morph_model
 
std::vector< size_t > m_fixed_upos
 
std::vector< std::shared_ptr< workbench_t > > m_workbenches
 
- Protected Attributes inherited from deeplima::eigen_impl::BiRnnInferenceBase
std::vector< std::shared_ptr< Op_Base > > m_ops
 
std::vector< std::shared_ptr< param_base_t > > m_params
 
std::vector< std::vector< std::shared_ptr< Op_Base::workbench_t > > > m_wb
 
uint_dicts_holder_t m_input_uint_dicts
 
std::vector< std::string > m_input_uint_dicts_names
 
str_dicts_holder_t m_input_str_dicts
 
std::vector< std::string > m_input_str_dicts_names
 
std::vector< std::vector< std::string > > m_output_str_dicts
 
std::vector< std::string > m_output_str_dicts_names
 
std::vector< params_bilstm_spec_t > m_lstm
 
std::map< std::string, size_t > m_lstm_idx
 
std::vector< std::shared_ptr< params_multilayer_bilstm_spec_t > > m_multi_bilstm
 
std::map< std::string, size_t > m_multi_bilstm_idx
 
std::vector< params_linear_t< Eigen::MatrixXf, Eigen::VectorXf > > m_linear
 
std::map< std::string, size_t > m_linear_idx
 

Additional Inherited Members

- Protected Types inherited from deeplima::eigen_impl::BiRnnInferenceBase
typedef params_bilstm_t< Eigen::MatrixXf, Eigen::VectorXf > params_bilstm_spec_t
 
typedef params_multilayer_bilstm_t< Eigen::MatrixXf, Eigen::VectorXf > params_multilayer_bilstm_spec_t
 

Detailed Description

Definition at line 44 of file lemmatization_eigen_inference_impl.h.

Member Typedef Documentation

◆ Parent

Constructor & Destructor Documentation

◆ ~BiRnnSeq2SeqEigenInferenceForLemmatization()

virtual deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::~BiRnnSeq2SeqEigenInferenceForLemmatization ( )
virtualdefault

Member Function Documentation

◆ convert_from_torch()

void deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::convert_from_torch ( const std::string &  fn)
overrideprotectedvirtual

Implements deeplima::eigen_impl::BiRnnInferenceBase.

Definition at line 26 of file lemmatization_model.cpp.

◆ get_class_names()

const std::vector< std::string > & deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::get_class_names ( ) const
inline

Definition at line 262 of file lemmatization_eigen_inference_impl.h.

◆ get_classes()

const std::vector< std::vector< std::string > > & deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::get_classes ( ) const
inline

Definition at line 257 of file lemmatization_eigen_inference_impl.h.

◆ get_embd_fn()

const std::string & deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::get_embd_fn ( size_t  idx) const
inline

Definition at line 272 of file lemmatization_eigen_inference_impl.h.

◆ get_fixed_upos()

const std::vector< size_t > & deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::get_fixed_upos ( ) const
inline

Definition at line 277 of file lemmatization_eigen_inference_impl.h.

◆ get_morph_model()

const morph_model::morph_model_t & deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::get_morph_model ( ) const
inline

Definition at line 267 of file lemmatization_eigen_inference_impl.h.

◆ get_precomputed_dim()

virtual size_t deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::get_precomputed_dim ( ) const
inlinevirtual

Definition at line 83 of file lemmatization_eigen_inference_impl.h.

◆ init_new_worker()

virtual size_t deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::init_new_worker ( size_t  input_len,
bool  precomputed_input = false 
)
inlineoverridevirtual

Reimplemented from deeplima::eigen_impl::BiRnnInferenceBase.

Definition at line 64 of file lemmatization_eigen_inference_impl.h.

◆ load()

virtual void deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::load ( const std::string &  fn)
inlineoverridevirtual

◆ precompute_inputs()

virtual void deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::precompute_inputs ( const Eigen::MatrixXf &  inputs,
Eigen::MatrixXf &  outputs,
int64_t  input_size 
)
inlineoverridevirtual

◆ predict() [1/2]

virtual void deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::predict ( size_t  worker_id,
const Eigen::MatrixXf &  inputs,
const Eigen::MatrixXf &  input_feats,
int64_t  input_len,
int64_t  output_max_len,
int64_t  ,
std::vector< uint32_t > &  output,
const std::vector< std::string > &   
)
inlinevirtual

Definition at line 127 of file lemmatization_eigen_inference_impl.h.

◆ predict() [2/2]

virtual void deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::predict ( size_t  ,
const Eigen::MatrixXf &  ,
int64_t  ,
int64_t  ,
int64_t  ,
int64_t  ,
std::shared_ptr< StdMatrix< uint8_t > > &  ,
const std::vector< std::string > &   
)
inlineoverridevirtual

Member Data Documentation

◆ m_class_names

std::vector<std::string> deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::m_class_names
protected

Definition at line 283 of file lemmatization_eigen_inference_impl.h.

◆ m_classes

std::vector<std::vector<std::string> > deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::m_classes
protected

Definition at line 284 of file lemmatization_eigen_inference_impl.h.

◆ m_embd_fn

std::vector<std::string> deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::m_embd_fn
protected

Definition at line 285 of file lemmatization_eigen_inference_impl.h.

◆ m_fixed_upos

std::vector<size_t> deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::m_fixed_upos
protected

Definition at line 287 of file lemmatization_eigen_inference_impl.h.

◆ m_morph_model

morph_model::morph_model_t deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::m_morph_model
protected

Definition at line 286 of file lemmatization_eigen_inference_impl.h.

◆ m_workbenches

std::vector<std::shared_ptr<workbench_t> > deeplima::lemmatization::eigen_impl::BiRnnSeq2SeqEigenInferenceForLemmatization::m_workbenches
protected

Definition at line 289 of file lemmatization_eigen_inference_impl.h.


The documentation for this class was generated from the following files: