diff options
author | Chris Dyer <cdyer@cs.cmu.edu> | 2011-07-06 19:54:58 -0400 |
---|---|---|
committer | Chris Dyer <cdyer@cs.cmu.edu> | 2011-07-06 19:54:58 -0400 |
commit | fe4b60f8669f0bdfcc67832e5487b33bd4b28938 (patch) | |
tree | d26226cf3ac885175e226d47b19f9ce8eea6a6c7 /decoder/ff_ngrams.h | |
parent | 164d32f02604ee5bff5de94ad669fb2b4d12d34a (diff) |
ngram count features
Diffstat (limited to 'decoder/ff_ngrams.h')
-rw-r--r-- | decoder/ff_ngrams.h | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/decoder/ff_ngrams.h b/decoder/ff_ngrams.h new file mode 100644 index 00000000..82f61b33 --- /dev/null +++ b/decoder/ff_ngrams.h @@ -0,0 +1,29 @@ +#ifndef _NGRAMS_FF_H_ +#define _NGRAMS_FF_H_ + +#include <vector> +#include <map> +#include <string> + +#include "ff.h" + +struct NgramDetectorImpl; +class NgramDetector : public FeatureFunction { + public: + // param = "filename.lm [-o n]" + NgramDetector(const std::string& param); + ~NgramDetector(); + virtual void FinalTraversalFeatures(const void* context, + SparseVector<double>* features) const; + 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* out_context) const; + private: + NgramDetectorImpl* pimpl_; +}; + +#endif |