Class BertEncoderClassifier

• Defined in File bert.h

Inheritance Relationships

Base Types

• public marian::EncoderClassifier (Class EncoderClassifier)

• public marian::data::RNGEngine (Class RNGEngine)

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)