diff options
Diffstat (limited to 'extractor/features/feature.h')
-rw-r--r-- | extractor/features/feature.h | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/extractor/features/feature.h b/extractor/features/feature.h new file mode 100644 index 00000000..aca58401 --- /dev/null +++ b/extractor/features/feature.h @@ -0,0 +1,36 @@ +#ifndef _FEATURE_H_ +#define _FEATURE_H_ + +#include <string> + +//TODO(pauldb): include headers nicely. +#include "../phrase.h" + +using namespace std; + +struct FeatureContext { + FeatureContext(const Phrase& source_phrase, const Phrase& target_phrase, + double source_phrase_count, int pair_count, int num_samples) : + source_phrase(source_phrase), target_phrase(target_phrase), + source_phrase_count(source_phrase_count), pair_count(pair_count), + num_samples(num_samples) {} + + Phrase source_phrase; + Phrase target_phrase; + double source_phrase_count; + int pair_count; + int num_samples; +}; + +class Feature { + public: + virtual double Score(const FeatureContext& context) const = 0; + + virtual string GetName() const = 0; + + virtual ~Feature(); + + static const double MAX_SCORE; +}; + +#endif |