Class Trainer

• Defined in File costs.h

Inheritance Relationships

Base Type

• public ICriterionFunction

Class Documentation

class Trainer : public ICriterionFunction

Public Functions

Trainer(Ptr<IModel> model, Ptr<ICost> cost)

virtual ~Trainer()

Ptr<IModel> getModel()

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

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

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

virtual void clear(Ptr<ExpressionGraph> graph)

Protected Attributes

Ptr<IModel> model_

Ptr<ICost> cost_