File size: 1,069 Bytes
158b61b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
#pragma once
#include <string>
#include <algorithm>
#include "VWFeatureContext.h"
#include "moses/Util.h"
namespace Moses
{
class VWFeatureContextWindow : public VWFeatureContext
{
public:
VWFeatureContextWindow(const std::string &line)
: VWFeatureContext(line, DEFAULT_WINDOW_SIZE) {
ReadParameters();
// Call this last
VWFeatureBase::UpdateRegister();
}
virtual void operator()(const InputType &input
, const Phrase &contextPhrase
, const AlignmentInfo &alignmentInfo
, Discriminative::Classifier &classifier
, Discriminative::FeatureVector &outFeatures) const {
for (size_t i = 0; i < m_contextSize; i++)
outFeatures.push_back(classifier.AddLabelIndependentFeature("tcwin^-" + SPrint(i + 1) + "^" + GetWord(contextPhrase, i)));
}
virtual void SetParameter(const std::string& key, const std::string& value) {
VWFeatureContext::SetParameter(key, value);
}
private:
static const int DEFAULT_WINDOW_SIZE = 1;
};
}
|