mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-09-11 11:25:40 +03:00
66 lines
1.8 KiB
C++
66 lines
1.8 KiB
C++
#include <iostream>
|
|
#include <fstream>
|
|
#include <assert.h>
|
|
#include <stdlib.h>
|
|
#include <string>
|
|
#include <queue>
|
|
#include <map>
|
|
#include <cmath>
|
|
|
|
#include "ScoreFeature.h"
|
|
#include "extract-ghkm/Node.h"
|
|
|
|
using namespace MosesTraining;
|
|
using namespace Moses;
|
|
using namespace GHKM;
|
|
|
|
namespace MosesTraining
|
|
{
|
|
|
|
|
|
class InternalStructFeature : public ScoreFeature
|
|
{
|
|
public:
|
|
InternalStructFeature() : m_type(0) {};
|
|
/** Add the values for this feature function. */
|
|
void add(const ScoreFeatureContext& context,
|
|
std::vector<float>& denseValues,
|
|
std::map<std::string,float>& sparseValues) const;
|
|
|
|
|
|
protected:
|
|
/** Overridden in subclass */
|
|
virtual void add(const std::string *treeFragment,
|
|
float count,
|
|
std::vector<float>& denseValues,
|
|
std::map<std::string,float>& sparseValues) const = 0;
|
|
int m_type;
|
|
|
|
};
|
|
|
|
class InternalStructFeatureDense : public InternalStructFeature
|
|
{
|
|
public:
|
|
InternalStructFeatureDense()
|
|
:InternalStructFeature(){m_type=1;} //std::cout<<"InternalStructFeatureDense: Construct "<<m_type<<"\n";}
|
|
protected:
|
|
virtual void add(const std::string *treeFragment,
|
|
float count,
|
|
std::vector<float>& denseValues,
|
|
std::map<std::string,float>& sparseValues) const;
|
|
};
|
|
|
|
class InternalStructFeatureSparse : public InternalStructFeature
|
|
{
|
|
public:
|
|
InternalStructFeatureSparse()
|
|
:InternalStructFeature(){m_type=2;}// std::cout<<"InternalStructFeatureSparse: Construct "<<m_type<<"\n";}
|
|
protected:
|
|
virtual void add(const std::string *treeFragment,
|
|
float count,
|
|
std::vector<float>& denseValues,
|
|
std::map<std::string,float>& sparseValues) const;
|
|
};
|
|
|
|
}
|