diff options
author | Rico Sennrich <rico.sennrich@gmx.ch> | 2014-09-17 20:40:11 +0400 |
---|---|---|
committer | Rico Sennrich <rico.sennrich@gmx.ch> | 2014-09-22 13:49:20 +0400 |
commit | d39cbca0b9532c8e51d1a42eef2cb2ef2a8d9bf7 (patch) | |
tree | 1b9299d981f765668f35b117c11a9fb3a69c0663 /mert/Data.cpp | |
parent | 3d00e5dc8ceb19d619e2ed2d0aa3f1b2a160944a (diff) |
(optionally) use n-best file for evaluator/return-best-dev
this adds support for metrics that rely on alignment / trees
Diffstat (limited to 'mert/Data.cpp')
-rw-r--r-- | mert/Data.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/mert/Data.cpp b/mert/Data.cpp index b6e3ed7f9..08e80409f 100644 --- a/mert/Data.cpp +++ b/mert/Data.cpp @@ -135,7 +135,7 @@ void Data::load(const std::string &featfile, const std::string &scorefile) m_score_data->load(scorefile); } -void Data::loadNBest(const string &file) +void Data::loadNBest(const string &file, bool oneBest) { TRACE_ERR("loading nbest from " << file << endl); util::FilePiece in(file.c_str()); @@ -154,6 +154,7 @@ void Data::loadNBest(const string &file) util::TokenIter<util::MultiCharacter> it(line, util::MultiCharacter("|||")); sentence_index = ParseInt(*it); + if (oneBest && m_score_data->exists(sentence_index)) continue; ++it; sentence = it->as_string(); ++it; |