diff options
| author | Chris Dyer <redpony@gmail.com> | 2013-11-13 11:22:24 -0800 |
|---|---|---|
| committer | Chris Dyer <redpony@gmail.com> | 2013-11-13 11:22:24 -0800 |
| commit | 9be8d89b5a4065a81b26d8af1f3443d152e7922a (patch) | |
| tree | 237090ff519a0419c3ba379ec3a6884f05caa6c2 /extractor/rule_factory.h | |
| parent | 8a24bb77bc2e9fd17a6f6529a2942cde96a6af49 (diff) | |
| parent | 4a9449a564e626fe004200b730bfaa44d6152e0f (diff) | |
Merge pull request #27 from pks/master
Tidying (soft) syntax features; loo for C++ extractor; updates for dtrain
Diffstat (limited to 'extractor/rule_factory.h')
| -rw-r--r-- | extractor/rule_factory.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/extractor/rule_factory.h b/extractor/rule_factory.h index 52e8712a..c7332720 100644 --- a/extractor/rule_factory.h +++ b/extractor/rule_factory.h @@ -3,6 +3,7 @@ #include <memory> #include <vector> +#include <unordered_set> #include "matchings_trie.h" @@ -71,7 +72,7 @@ class HieroCachingRuleFactory { // Constructs SCFG rules for a given sentence. // (See class description for more details.) - virtual Grammar GetGrammar(const vector<int>& word_ids); + virtual Grammar GetGrammar(const vector<int>& word_ids, const unordered_set<int> blacklisted_sentence_ids, const shared_ptr<DataArray> source_data_array); protected: HieroCachingRuleFactory(); |