blob: 86701add3772304e8f2bfd623717d9f385b988d8 (
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
25
26
|
#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 std::string* extra = NULL) const;
void InitVector(std::vector<double>* w) const;
void InitSparseVector(SparseVector<double>* w) const;
void InitFromVector(const std::vector<double>& w);
void InitFromVector(const SparseVector<double>& w);
void SetWeight(SparseVector<double>* v, const std::string f, const double w);
void SetWeight(SparseVector<double>* v, const WordID fid, const double w);
std::vector<double>* getw() { return &wv_; }; // probably a hack
void sz();
private:
std::vector<double> wv_;
};
#endif
|
