summaryrefslogtreecommitdiff
path: root/utils/weights.h
blob: 7664810b49d389db52f0fb97d3742997009235e5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#ifndef _WEIGHTS_H_
#define _WEIGHTS_H_

#include <string>
#include <vector>
#include "sparse_vector.h"

// warning: in the future this will become float
typedef double weight_t;

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 std::string* extra = NULL) const;
  void InitVector(std::vector<weight_t>* w) const;
  void InitSparseVector(SparseVector<weight_t>* w) const;
  void InitFromVector(const std::vector<weight_t>& w);
  void InitFromVector(const SparseVector<weight_t>& w);
 private:
  std::vector<weight_t> wv_;
};

#endif