From fed4a267a202a9d60326861b0a71205503010d24 Mon Sep 17 00:00:00 2001 From: Waleed Ammar Date: Sat, 16 Jun 2012 05:35:49 -0400 Subject: enable regex-based feature templates --- decoder/ff_context.h | 29 ++++++++++++++++++++++++++--- 1 file changed, 26 insertions(+), 3 deletions(-) (limited to 'decoder/ff_context.h') diff --git a/decoder/ff_context.h b/decoder/ff_context.h index 0d22b027..89bcb557 100644 --- a/decoder/ff_context.h +++ b/decoder/ff_context.h @@ -1,23 +1,46 @@ + #ifndef _FF_CONTEXT_H_ #define _FF_CONTEXT_H_ #include +#include #include "ff.h" +using namespace boost::xpressive; +using namespace std; + class RuleContextFeatures : public FeatureFunction { public: - RuleContextFeatures(const std::string& param); + RuleContextFeatures(const string& param); protected: virtual void TraversalFeaturesImpl(const SentenceMetadata& smeta, const Hypergraph::Edge& edge, - const std::vector& ant_contexts, + const vector& ant_contexts, SparseVector* features, SparseVector* estimated_features, void* context) const; virtual void PrepareForInput(const SentenceMetadata& smeta); + virtual void ParseArgs(const string& in); + virtual string Escape(const string& x) const; + virtual void ReplaceMacroWithString(string& feature_instance, + bool token_vs_label, + int relative_location, + const string& actual_token) const; + virtual void ReplaceTokenMacroWithString(string& feature_instance, + int relative_location, + const string& actual_token) const; + virtual void ReplaceLabelMacroWithString(string& feature_instance, + int relative_location, + const string& actual_token) const; + virtual void Error(const string&) const; + private: - std::vector current_input; + vector token_relative_locations, label_relative_locations; + string feature_template; + vector current_input; WordID kSOS, kEOS; + sregex macro_regex; + }; #endif -- cgit v1.2.3