blob: 0d22b02713de4d9fc6a9de487e18e02d191a4af0 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
#ifndef _FF_CONTEXT_H_
#define _FF_CONTEXT_H_
#include <vector>
#include "ff.h"
class RuleContextFeatures : public FeatureFunction {
public:
RuleContextFeatures(const std::string& param);
protected:
virtual void TraversalFeaturesImpl(const SentenceMetadata& smeta,
const Hypergraph::Edge& edge,
const std::vector<const void*>& ant_contexts,
SparseVector<double>* features,
SparseVector<double>* estimated_features,
void* context) const;
virtual void PrepareForInput(const SentenceMetadata& smeta);
private:
std::vector<WordID> current_input;
WordID kSOS, kEOS;
};
#endif
|
