summaryrefslogtreecommitdiff
path: root/decoder/weights.h
diff options
context:
space:
mode:
Diffstat (limited to 'decoder/weights.h')
-rw-r--r--decoder/weights.h21
1 files changed, 21 insertions, 0 deletions
diff --git a/decoder/weights.h b/decoder/weights.h
new file mode 100644
index 00000000..f19aa3ce
--- /dev/null
+++ b/decoder/weights.h
@@ -0,0 +1,21 @@
+#ifndef _WEIGHTS_H_
+#define _WEIGHTS_H_
+
+#include <string>
+#include <map>
+#include <vector>
+#include "sparse_vector.h"
+
+class Weights {
+ public:
+ Weights() {}
+ void InitFromFile(const std::string& fname, std::vector<std::string>* feature_list = NULL);
+ void WriteToFile(const std::string& fname, bool hide_zero_value_features = true) const;
+ void InitVector(std::vector<double>* w) const;
+ void InitSparseVector(SparseVector<double>* w) const;
+ void InitFromVector(const std::vector<double>& w);
+ private:
+ std::vector<double> wv_;
+};
+
+#endif