Class BertEncoderClassifier

Inheritance Relationships

Base Types

Class Documentation

class BertEncoderClassifier : public marian::EncoderClassifier, public marian::data::RNGEngine

BERT-specific version of EncoderClassifier, mostly here to automatically convert a CorpusBatch to BertBatch.

Public Functions

BertEncoderClassifier(Ptr<Options> options)
std::vector<Ptr<ClassifierState>> apply(Ptr<ExpressionGraph> graph, Ptr<data::CorpusBatch> batch, bool clearGraph)
std::vector<Ptr<ClassifierState>> apply(Ptr<ExpressionGraph> graph, Ptr<data::BertBatch> bertBatch, bool clearGraph)