Model.h
Go to the documentation of this file.
1 #pragma once
2 
3 #include <memory>
4 #include <string>
5 #include <utility>
6 #include <vector>
7 
11 
12 namespace LSTME
13 {
14 
15 class Model
16 {
17 private:
19 
20  static const std::vector<InputConfigKeys> sliceInputs;
21  static const std::vector<InputConfigKeys> prongInputs;
22  static const std::vector<std::string> outputs;
23 
24 public:
25  Model(const std::string &savedir);
26  LSTMEnergy predict(const VarDict &varDict);
27 };
28 
29 }
static const std::vector< std::string > outputs
Definition: Model.h:22
SimpleModel simpleModel
Definition: Model.h:18
static const std::vector< InputConfigKeys > sliceInputs
Definition: Model.h:20
LSTMEnergy predict(const VarDict &varDict)
Definition: Model.cxx:27
Model(const std::string &savedir)
Definition: Model.cxx:23
static const std::vector< InputConfigKeys > prongInputs
Definition: Model.h:21
Definition: VarDict.h:7
enum BeamMode string