Class ScorerWrapper

Inheritance Relationships

Base Type

Class Documentation

class ScorerWrapper : public marian::Scorer

Public Functions

ScorerWrapper(Ptr<models::IModel> encdec, const std::string &name, float weight, std::vector<io::Item> &items)
ScorerWrapper(Ptr<models::IModel> encdec, const std::string &name, float weight, const std::string &fname)
ScorerWrapper(Ptr<models::IModel> encdec, const std::string &name, float weight, const void *ptr)
virtual ~ScorerWrapper()
virtual void init(Ptr<ExpressionGraph> graph)
virtual void clear(Ptr<ExpressionGraph> graph)
virtual Ptr<ScorerState> startState(Ptr<ExpressionGraph> graph, Ptr<data::CorpusBatch> batch)
virtual Ptr<ScorerState> step(Ptr<ExpressionGraph> graph, Ptr<ScorerState> state, const std::vector<IndexType> &hypIndices, const Words &words, const std::vector<IndexType> &batchIndices, int beamSize)
virtual void setShortlistGenerator(Ptr<const data::ShortlistGenerator> shortlistGenerator)
virtual Ptr<data::Shortlist> getShortlist()
virtual std::vector<float> getAlignment()