diff options
author | Hieu Hoang <hieuhoang@gmail.com> | 2012-11-12 23:56:18 +0400 |
---|---|---|
committer | Hieu Hoang <hieuhoang@gmail.com> | 2012-11-12 23:56:18 +0400 |
commit | 5e3ef23cef6101d2c098eb3445f562e8f595655b (patch) | |
tree | b8c332b6fa82bae84ea4910967a10ba1b08a7107 /moses/PartialTranslOptColl.h | |
parent | 8c785cff2b1be3cccd76ea9026f71b649762dfc3 (diff) |
move moses/src/* to moses/
Diffstat (limited to 'moses/PartialTranslOptColl.h')
-rw-r--r-- | moses/PartialTranslOptColl.h | 83 |
1 files changed, 83 insertions, 0 deletions
diff --git a/moses/PartialTranslOptColl.h b/moses/PartialTranslOptColl.h new file mode 100644 index 000000000..bbe4d05a1 --- /dev/null +++ b/moses/PartialTranslOptColl.h @@ -0,0 +1,83 @@ +// $Id$ + +/*********************************************************************** +Moses - factored phrase-based language decoder +Copyright (C) 2006 University of Edinburgh + +This library is free software; you can redistribute it and/or +modify it under the terms of the GNU Lesser General Public +License as published by the Free Software Foundation; either +version 2.1 of the License, or (at your option) any later version. + +This library is distributed in the hope that it will be useful, +but WITHOUT ANY WARRANTY; without even the implied warranty of +MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +Lesser General Public License for more details. + +You should have received a copy of the GNU Lesser General Public +License along with this library; if not, write to the Free Software +Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA +***********************************************************************/ + +#ifndef moses_PartialTranslOptColl_h +#define moses_PartialTranslOptColl_h + +#include <list> +#include <iostream> +#include "TranslationOption.h" +#include "Util.h" +#include "StaticData.h" +#include "FactorTypeSet.h" + +namespace Moses +{ + +/** Contains partial translation options, while these are constructed in the class TranslationOption. + * The factored translation model allows for multiple translation and + * generation steps during a single Hypothesis expansion. For efficiency, + * all these expansions are precomputed and stored as TranslationOption. + * The expansion process itself may be still explode, so efficient handling + * of partial translation options during expansion is required. + * This class assists in this tasks by implementing pruning. + * This implementation is similar to the one in HypothesisStack. + */ +class PartialTranslOptColl +{ +protected: + std::vector<TranslationOption*> m_list; + float m_bestScore; /**< score of the best translation option */ + float m_worstScore; /**< score of the worse translation option */ + size_t m_maxSize; /**< maximum number of translation options allowed */ + size_t m_totalPruned; /**< number of options pruned */ + +public: + PartialTranslOptColl(); + + /** destructor, cleans out list */ + ~PartialTranslOptColl() { + RemoveAllInColl( m_list ); + } + + void AddNoPrune(const TranslationSystem* system, TranslationOption *partialTranslOpt); + void Add(const TranslationSystem* system, TranslationOption *partialTranslOpt); + void Prune(); + + /** returns list of translation options */ + const std::vector<TranslationOption*>& GetList() const { + return m_list; + } + + /** clear out the list */ + void DetachAll() { + m_list.clear(); + } + + /** return number of pruned partial hypotheses */ + size_t GetPrunedCount() { + return m_totalPruned; + } + +}; + +} +#endif |