Class EncoderClassifier

• Defined in File encoder_classifier.h

Inheritance Relationships

Base Type

• public marian::EncoderClassifierBase (Class EncoderClassifierBase)

Derived Type

• public marian::BertEncoderClassifier (Class BertEncoderClassifier)

Class Documentation

class EncoderClassifier : public marian::EncoderClassifierBase

Subclassed by marian::BertEncoderClassifier

Public Types

typedef data::Corpus dataset_type

Public Functions

EncoderClassifier(Ptr<Options> options)

virtual Ptr<Options> getOptions()

std::vector<Ptr<EncoderBase>> &getEncoders()

std::vector<Ptr<ClassifierBase>> &getClassifiers()

void push_back(Ptr<EncoderBase> encoder)

void push_back(Ptr<ClassifierBase> classifier)

void load(Ptr<ExpressionGraph> graph, const std::string &name, bool markedReloaded)

void mmap(Ptr<ExpressionGraph> graph, const void *ptr, bool markedReloaded)

void save(Ptr<ExpressionGraph> graph, const std::string &name, bool)

void clear(Ptr<ExpressionGraph> graph)

template<typename T>
T opt(const std::string &key)

template<typename T>
T opt(const std::string &key, const T &def)

template<typename T>
void set(std::string key, T value)

virtual std::vector<Ptr<ClassifierState>> apply(Ptr<ExpressionGraph> graph, Ptr<data::CorpusBatch> batch, bool clearGraph)

virtual Logits build(Ptr<ExpressionGraph> graph, Ptr<data::CorpusBatch> batch, bool clearGraph = true)

virtual Logits build(Ptr<ExpressionGraph> graph, Ptr<data::Batch> batch, bool clearGraph = true)

Protected Functions

Config::YamlNode getModelParameters()

std::string getModelParametersAsString()

Protected Attributes

Ptr<Options> options_

std::string prefix_

std::vector<Ptr<EncoderBase>> encoders_

std::vector<Ptr<ClassifierBase>> classifiers_

bool inference_ = {false}

std::set<std::string> modelFeatures_