Blame view
src/rnnlm/rnnlm-lattice-rescoring.h
2.4 KB
8dcb6dfcb first commit |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 |
// rnnlm/rnnlm-lattice-rescoring.h // // Copyright 2017 Johns Hopkins University (author: Daniel Povey) // 2017 Yiming Wang // 2017 Hainan Xu // // See ../../COPYING for clarification regarding multiple authors // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // THIS CODE IS PROVIDED *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY // KIND, EITHER EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION ANY IMPLIED // WARRANTIES OR CONDITIONS OF TITLE, FITNESS FOR A PARTICULAR PURPOSE, // MERCHANTABLITY OR NON-INFRINGEMENT. // See the Apache 2 License for the specific language governing permissions and // limitations under the License. #ifndef KALDI_RNNLM_RNNLM_LATTICE_RESCORING_H_ #define KALDI_RNNLM_RNNLM_LATTICE_RESCORING_H_ #include <string> #include <vector> #include "base/kaldi-common.h" #include "fstext/deterministic-fst.h" #include "rnnlm/rnnlm-compute-state.h" #include "util/common-utils.h" namespace kaldi { namespace rnnlm { class KaldiRnnlmDeterministicFst : public fst::DeterministicOnDemandFst<fst::StdArc> { public: typedef fst::StdArc::Weight Weight; typedef fst::StdArc::StateId StateId; typedef fst::StdArc::Label Label; // Does not take ownership. KaldiRnnlmDeterministicFst(int32 max_ngram_order, const RnnlmComputeStateInfo &info); ~KaldiRnnlmDeterministicFst(); void Clear(); // We cannot use "const" because the pure virtual function in the interface is // not const. virtual StateId Start() { return start_state_; } // We cannot use "const" because the pure virtual function in the interface is // not const. virtual Weight Final(StateId s); virtual bool GetArc(StateId s, Label ilabel, fst::StdArc* oarc); private: typedef unordered_map <std::vector<Label>, StateId, VectorHasher<Label> > MapType; StateId start_state_; int32 max_ngram_order_; int32 bos_index_; int32 eos_index_; MapType wseq_to_state_; // Mapping from state-id to history sequence> std::vector<std::vector<Label> > state_to_wseq_; // Mapping from state-id to RNNLM states. // The pointers are owned in this class std::vector<RnnlmComputeState*> state_to_rnnlm_state_; }; } // namespace rnnlm } // namespace kaldi #endif // KALDI_RNNLM_RNNLM_LATTICE_RESCORING_H_ |