Class BeamSearch

• Defined in File beam_search.h

Class Documentation

class BeamSearch

Public Functions

BeamSearch(Ptr<Options> options, const std::vector<Ptr<Scorer>> &scorers, const Ptr<const Vocab> trgVocab)

Beams toHyps(const std::vector<unsigned int> &nBestKeys, const std::vector<float> &nBestPathScores, const size_t nBestBeamSize, const size_t vocabSize, const Beams &beams, const std::vector<Ptr<ScorerState>> &states, Ptr<data::CorpusBatch> batch, Ptr<class FactoredVocab> factoredVocab, size_t factorGroup, const std::vector<bool> &dropBatchEntries, const std::vector<IndexType> &batchIdxMap) const

std::vector<float> getAlignmentsForHypothesis(const std::vector<float> alignAll, Ptr<data::CorpusBatch> batch, int beamHypIdx, int currentBatchIdx, int origBatchIdx, int currentDimBatch) const

Beams purgeBeams(const Beams &beams, std::vector<IndexType> &batchIdxMap)

Histories search(Ptr<ExpressionGraph> graph, Ptr<data::CorpusBatch> batch)