2008-06-11 14:52:57 +04:00
|
|
|
// $Id$
|
2010-09-17 18:25:08 +04:00
|
|
|
#include <vector>
|
2013-09-10 17:36:21 +04:00
|
|
|
#include "util/exception.hh"
|
2008-06-11 14:52:57 +04:00
|
|
|
#include "ScoreComponentCollection.h"
|
|
|
|
#include "StaticData.h"
|
|
|
|
|
2010-10-07 02:06:49 +04:00
|
|
|
using namespace std;
|
|
|
|
|
2008-10-09 03:51:26 +04:00
|
|
|
namespace Moses
|
|
|
|
{
|
2013-09-08 18:43:18 +04:00
|
|
|
void ScorePair::PlusEquals(const ScorePair &other)
|
|
|
|
{
|
2013-09-27 12:35:24 +04:00
|
|
|
PlusEquals(other.denseScores);
|
|
|
|
std::map<StringPiece, float>::const_iterator iter;
|
|
|
|
for (iter = other.sparseScores.begin(); iter != other.sparseScores.end(); ++iter) {
|
|
|
|
PlusEquals(iter->first, iter->second);
|
|
|
|
}
|
2013-09-08 18:43:18 +04:00
|
|
|
}
|
|
|
|
|
2013-09-08 19:02:59 +04:00
|
|
|
void ScorePair::PlusEquals(const StringPiece &key, float value)
|
2013-09-08 18:43:18 +04:00
|
|
|
{
|
2013-09-27 12:35:24 +04:00
|
|
|
std::map<StringPiece, float>::iterator iter;
|
|
|
|
iter = sparseScores.find(key);
|
|
|
|
if (iter == sparseScores.end()) {
|
|
|
|
sparseScores[key] = value;
|
|
|
|
} else {
|
|
|
|
float &existingval = iter->second;
|
|
|
|
existingval += value;
|
|
|
|
}
|
2013-09-08 18:43:18 +04:00
|
|
|
}
|
2011-11-09 21:16:02 +04:00
|
|
|
|
2013-09-10 17:36:21 +04:00
|
|
|
std::ostream& operator<<(std::ostream& os, const ScorePair& rhs)
|
|
|
|
{
|
2013-09-27 12:35:24 +04:00
|
|
|
for (size_t i = 0; i < rhs.denseScores.size(); ++i) {
|
|
|
|
os << rhs.denseScores[i] << ",";
|
|
|
|
}
|
2013-09-10 17:36:21 +04:00
|
|
|
|
2013-09-27 12:35:24 +04:00
|
|
|
std::map<StringPiece, float>::const_iterator iter;
|
|
|
|
for (iter = rhs.sparseScores.begin(); iter != rhs.sparseScores.end(); ++iter) {
|
|
|
|
os << iter->first << "=" << iter->second << ",";
|
|
|
|
}
|
2013-09-10 17:36:21 +04:00
|
|
|
|
2013-09-27 12:35:24 +04:00
|
|
|
return os;
|
2013-09-10 17:36:21 +04:00
|
|
|
}
|
|
|
|
|
2011-11-09 21:16:02 +04:00
|
|
|
ScoreComponentCollection::ScoreIndexMap ScoreComponentCollection::s_scoreIndexes;
|
|
|
|
size_t ScoreComponentCollection::s_denseVectorSize = 0;
|
|
|
|
|
2013-09-25 03:13:49 +04:00
|
|
|
ScoreComponentCollection::
|
2013-09-27 12:35:24 +04:00
|
|
|
ScoreComponentCollection()
|
2013-09-25 03:13:49 +04:00
|
|
|
: m_scores(s_denseVectorSize)
|
2008-06-11 14:52:57 +04:00
|
|
|
{}
|
|
|
|
|
2010-09-28 19:13:50 +04:00
|
|
|
|
2013-09-27 12:35:24 +04:00
|
|
|
void
|
2013-09-25 03:13:49 +04:00
|
|
|
ScoreComponentCollection::
|
|
|
|
RegisterScoreProducer(const FeatureFunction* scoreProducer)
|
2011-11-09 21:16:02 +04:00
|
|
|
{
|
|
|
|
size_t start = s_denseVectorSize;
|
|
|
|
size_t end = start + scoreProducer->GetNumScoreComponents();
|
2013-09-27 12:35:24 +04:00
|
|
|
VERBOSE(1, "FeatureFunction: "
|
|
|
|
<< scoreProducer->GetScoreProducerDescription()
|
|
|
|
<< " start: " << start << " end: " << (end-1) << endl);
|
2011-11-09 21:16:02 +04:00
|
|
|
s_scoreIndexes[scoreProducer] = pair<size_t,size_t>(start,end);
|
|
|
|
s_denseVectorSize = end;
|
|
|
|
}
|
|
|
|
|
2011-11-16 15:49:31 +04:00
|
|
|
|
2013-09-27 12:35:24 +04:00
|
|
|
float
|
2013-09-25 03:13:49 +04:00
|
|
|
ScoreComponentCollection::
|
|
|
|
GetWeightedScore() const
|
2010-04-08 21:16:10 +04:00
|
|
|
{
|
2013-05-29 21:16:15 +04:00
|
|
|
return m_scores.inner_product(StaticData::Instance().GetAllWeights().m_scores);
|
2010-04-08 21:16:10 +04:00
|
|
|
}
|
2011-02-24 16:14:42 +03:00
|
|
|
|
2010-09-17 19:20:55 +04:00
|
|
|
void ScoreComponentCollection::MultiplyEquals(float scalar)
|
|
|
|
{
|
2013-05-29 21:16:15 +04:00
|
|
|
m_scores *= scalar;
|
2010-09-17 19:20:55 +04:00
|
|
|
}
|
2010-10-28 16:41:33 +04:00
|
|
|
|
2011-11-24 23:27:12 +04:00
|
|
|
// Multiply all weights of this sparse producer by a given scalar
|
2013-05-29 21:16:15 +04:00
|
|
|
void ScoreComponentCollection::MultiplyEquals(const FeatureFunction* sp, float scalar)
|
|
|
|
{
|
2012-01-09 16:04:18 +04:00
|
|
|
std::string prefix = sp->GetScoreProducerDescription() + FName::SEP;
|
2011-11-24 23:27:12 +04:00
|
|
|
for(FVector::FNVmap::const_iterator i = m_scores.cbegin(); i != m_scores.cend(); i++) {
|
|
|
|
std::stringstream name;
|
|
|
|
name << i->first;
|
|
|
|
if (name.str().substr( 0, prefix.length() ).compare( prefix ) == 0)
|
2013-05-29 21:16:15 +04:00
|
|
|
m_scores[i->first] = i->second * scalar;
|
2011-11-24 23:27:12 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-03-01 16:22:12 +04:00
|
|
|
// Count weights belonging to this sparse producer
|
2013-05-29 21:16:15 +04:00
|
|
|
size_t ScoreComponentCollection::GetNumberWeights(const FeatureFunction* sp)
|
|
|
|
{
|
2012-03-01 16:22:12 +04:00
|
|
|
std::string prefix = sp->GetScoreProducerDescription() + FName::SEP;
|
|
|
|
size_t weights = 0;
|
|
|
|
for(FVector::FNVmap::const_iterator i = m_scores.cbegin(); i != m_scores.cend(); i++) {
|
|
|
|
std::stringstream name;
|
|
|
|
name << i->first;
|
|
|
|
if (name.str().substr( 0, prefix.length() ).compare( prefix ) == 0)
|
2013-05-29 21:16:15 +04:00
|
|
|
weights++;
|
2012-03-01 16:22:12 +04:00
|
|
|
}
|
|
|
|
return weights;
|
|
|
|
}
|
|
|
|
|
2010-10-28 16:41:33 +04:00
|
|
|
void ScoreComponentCollection::DivideEquals(float scalar)
|
|
|
|
{
|
2013-05-29 21:16:15 +04:00
|
|
|
m_scores /= scalar;
|
2010-10-28 16:41:33 +04:00
|
|
|
}
|
2011-08-06 18:10:43 +04:00
|
|
|
|
2012-03-14 21:59:59 +04:00
|
|
|
void ScoreComponentCollection::CoreDivideEquals(float scalar)
|
|
|
|
{
|
2013-05-29 21:16:15 +04:00
|
|
|
m_scores.coreDivideEquals(scalar);
|
2012-03-14 21:59:59 +04:00
|
|
|
}
|
|
|
|
|
2012-03-13 20:34:48 +04:00
|
|
|
void ScoreComponentCollection::DivideEquals(const ScoreComponentCollection& rhs)
|
|
|
|
{
|
2013-05-29 21:16:15 +04:00
|
|
|
m_scores.divideEquals(rhs.m_scores);
|
2012-03-13 20:34:48 +04:00
|
|
|
}
|
|
|
|
|
2010-09-17 19:33:21 +04:00
|
|
|
void ScoreComponentCollection::MultiplyEquals(const ScoreComponentCollection& rhs)
|
|
|
|
{
|
2013-05-29 21:16:15 +04:00
|
|
|
m_scores *= rhs.m_scores;
|
2010-09-17 19:33:21 +04:00
|
|
|
}
|
|
|
|
|
2012-05-18 21:58:42 +04:00
|
|
|
void ScoreComponentCollection::MultiplyEqualsBackoff(const ScoreComponentCollection& rhs, float backoff)
|
2012-05-15 20:49:54 +04:00
|
|
|
{
|
2012-05-18 21:58:42 +04:00
|
|
|
m_scores.multiplyEqualsBackoff(rhs.m_scores, backoff);
|
|
|
|
}
|
|
|
|
|
|
|
|
void ScoreComponentCollection::MultiplyEquals(float core_r0, float sparse_r0)
|
|
|
|
{
|
|
|
|
m_scores.multiplyEquals(core_r0, sparse_r0);
|
2012-05-15 20:49:54 +04:00
|
|
|
}
|
|
|
|
|
2010-09-16 11:06:42 +04:00
|
|
|
std::ostream& operator<<(std::ostream& os, const ScoreComponentCollection& rhs)
|
|
|
|
{
|
2013-05-29 21:16:15 +04:00
|
|
|
os << rhs.m_scores;
|
|
|
|
return os;
|
2010-09-16 11:06:42 +04:00
|
|
|
}
|
2013-05-29 21:16:15 +04:00
|
|
|
void ScoreComponentCollection::L1Normalise()
|
|
|
|
{
|
2012-04-26 02:48:12 +04:00
|
|
|
m_scores /= m_scores.l1norm_coreFeatures();
|
2010-09-28 19:13:50 +04:00
|
|
|
}
|
|
|
|
|
2013-05-29 21:16:15 +04:00
|
|
|
float ScoreComponentCollection::GetL1Norm() const
|
|
|
|
{
|
2011-05-31 19:39:48 +04:00
|
|
|
return m_scores.l1norm();
|
|
|
|
}
|
|
|
|
|
2013-05-29 21:16:15 +04:00
|
|
|
float ScoreComponentCollection::GetL2Norm() const
|
|
|
|
{
|
2011-04-23 16:13:43 +04:00
|
|
|
return m_scores.l2norm();
|
|
|
|
}
|
|
|
|
|
2013-05-29 21:16:15 +04:00
|
|
|
float ScoreComponentCollection::GetLInfNorm() const
|
|
|
|
{
|
2011-12-23 14:55:54 +04:00
|
|
|
return m_scores.linfnorm();
|
|
|
|
}
|
|
|
|
|
2013-05-29 21:16:15 +04:00
|
|
|
size_t ScoreComponentCollection::L1Regularize(float lambda)
|
|
|
|
{
|
2012-07-26 20:32:50 +04:00
|
|
|
return m_scores.l1regularize(lambda);
|
|
|
|
}
|
|
|
|
|
2013-05-29 21:16:15 +04:00
|
|
|
void ScoreComponentCollection::L2Regularize(float lambda)
|
|
|
|
{
|
2012-07-26 20:32:50 +04:00
|
|
|
m_scores.l2regularize(lambda);
|
|
|
|
}
|
|
|
|
|
2013-05-29 21:16:15 +04:00
|
|
|
size_t ScoreComponentCollection::SparseL1Regularize(float lambda)
|
|
|
|
{
|
2012-06-01 04:49:42 +04:00
|
|
|
return m_scores.sparseL1regularize(lambda);
|
2012-05-10 21:54:24 +04:00
|
|
|
}
|
|
|
|
|
2013-05-29 21:16:15 +04:00
|
|
|
void ScoreComponentCollection::SparseL2Regularize(float lambda)
|
|
|
|
{
|
2012-06-01 04:49:42 +04:00
|
|
|
m_scores.sparseL2regularize(lambda);
|
2012-05-10 21:54:24 +04:00
|
|
|
}
|
|
|
|
|
2014-08-06 14:38:51 +04:00
|
|
|
void ScoreComponentCollection::Save(ostream& out, bool multiline) const
|
2013-05-29 21:16:15 +04:00
|
|
|
{
|
2014-08-06 14:38:51 +04:00
|
|
|
string sep = " ";
|
|
|
|
string linesep = "\n";
|
|
|
|
if (!multiline) {
|
|
|
|
sep = "=";
|
|
|
|
linesep = " ";
|
|
|
|
}
|
2011-11-16 13:13:17 +04:00
|
|
|
ScoreIndexMap::const_iterator iter = s_scoreIndexes.begin();
|
|
|
|
for (; iter != s_scoreIndexes.end(); ++iter ) {
|
2011-11-16 15:49:31 +04:00
|
|
|
string name = iter->first->GetScoreProducerDescription();
|
|
|
|
IndexPair ip = iter->second; // feature indices
|
2011-11-17 18:55:10 +04:00
|
|
|
if (ip.second-ip.first == 1) {
|
2014-08-06 14:38:51 +04:00
|
|
|
out << name << sep << m_scores[ip.first] << linesep;
|
2011-11-17 18:55:10 +04:00
|
|
|
} else {
|
|
|
|
for (size_t i=ip.first; i < ip.second; ++i) {
|
|
|
|
ostringstream fullname;
|
|
|
|
fullname << name << "_" << (i + 1 - ip.first);
|
2014-08-06 14:38:51 +04:00
|
|
|
out << fullname.str() << sep << m_scores[i] << linesep;
|
2011-11-17 18:55:10 +04:00
|
|
|
}
|
2011-11-16 15:49:31 +04:00
|
|
|
}
|
2011-11-16 13:13:17 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
// write sparse features
|
2014-08-06 14:38:51 +04:00
|
|
|
m_scores.write(out,sep);
|
2011-11-16 15:49:31 +04:00
|
|
|
}
|
|
|
|
|
2013-05-29 21:16:15 +04:00
|
|
|
void ScoreComponentCollection::Save(const string& filename) const
|
|
|
|
{
|
2011-11-16 15:49:31 +04:00
|
|
|
ofstream out(filename.c_str());
|
|
|
|
if (!out) {
|
|
|
|
ostringstream msg;
|
|
|
|
msg << "Unable to open " << filename;
|
|
|
|
throw runtime_error(msg.str());
|
|
|
|
}
|
|
|
|
Save(out);
|
2011-11-16 13:13:17 +04:00
|
|
|
out.close();
|
|
|
|
}
|
|
|
|
|
2013-09-27 12:35:24 +04:00
|
|
|
void
|
2013-09-25 03:13:49 +04:00
|
|
|
ScoreComponentCollection::
|
2014-06-06 23:35:58 +04:00
|
|
|
Assign(const FeatureFunction* sp, const string &line)
|
2013-05-29 21:16:15 +04:00
|
|
|
{
|
2011-09-20 19:32:26 +04:00
|
|
|
istringstream istr(line);
|
|
|
|
while(istr) {
|
|
|
|
string namestring;
|
|
|
|
FValue value;
|
|
|
|
istr >> namestring;
|
|
|
|
if (!istr) break;
|
|
|
|
istr >> value;
|
|
|
|
FName fname(sp->GetScoreProducerDescription(), namestring);
|
|
|
|
m_scores[fname] = value;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-09-27 12:35:24 +04:00
|
|
|
void
|
2013-09-25 03:13:49 +04:00
|
|
|
ScoreComponentCollection::
|
2013-09-27 12:35:24 +04:00
|
|
|
Assign(const FeatureFunction* sp, const std::vector<float>& scores)
|
2013-09-25 03:13:49 +04:00
|
|
|
{
|
2013-09-10 17:36:21 +04:00
|
|
|
IndexPair indexes = GetIndexes(sp);
|
|
|
|
size_t numScores = indexes.second - indexes.first;
|
|
|
|
|
|
|
|
if (scores.size() != numScores) {
|
2013-09-27 12:35:24 +04:00
|
|
|
UTIL_THROW(util::Exception, "Feature function "
|
|
|
|
<< sp->GetScoreProducerDescription() << " specified "
|
|
|
|
<< numScores << " dense scores or weights. Actually has "
|
|
|
|
<< scores.size());
|
2013-09-10 17:36:21 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
for (size_t i = 0; i < scores.size(); ++i) {
|
|
|
|
m_scores[i + indexes.first] = scores[i];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2013-07-31 15:32:53 +04:00
|
|
|
void ScoreComponentCollection::InvertDenseFeatures(const FeatureFunction* sp)
|
|
|
|
{
|
|
|
|
|
|
|
|
Scores old_scores = GetScoresForProducer(sp);
|
|
|
|
Scores new_scores(old_scores.size());
|
|
|
|
|
|
|
|
for (size_t i = 0; i != old_scores.size(); ++i) {
|
|
|
|
new_scores[i] = -old_scores[i];
|
|
|
|
}
|
|
|
|
|
|
|
|
Assign(sp, new_scores);
|
|
|
|
}
|
|
|
|
|
2013-05-13 16:19:25 +04:00
|
|
|
void ScoreComponentCollection::ZeroDenseFeatures(const FeatureFunction* sp)
|
|
|
|
{
|
|
|
|
size_t numScores = sp->GetNumScoreComponents();
|
|
|
|
Scores vec(numScores, 0);
|
|
|
|
|
|
|
|
Assign(sp, vec);
|
|
|
|
}
|
|
|
|
|
2013-05-15 14:37:21 +04:00
|
|
|
//! get subset of scores that belong to a certain sparse ScoreProducer
|
|
|
|
FVector ScoreComponentCollection::GetVectorForProducer(const FeatureFunction* sp) const
|
|
|
|
{
|
|
|
|
FVector fv(s_denseVectorSize);
|
|
|
|
std::string prefix = sp->GetScoreProducerDescription() + FName::SEP;
|
|
|
|
for(FVector::FNVmap::const_iterator i = m_scores.cbegin(); i != m_scores.cend(); i++) {
|
|
|
|
std::stringstream name;
|
|
|
|
name << i->first;
|
|
|
|
if (name.str().substr( 0, prefix.length() ).compare( prefix ) == 0)
|
|
|
|
fv[i->first] = i->second;
|
|
|
|
}
|
|
|
|
return fv;
|
|
|
|
}
|
|
|
|
|
2013-09-08 17:57:31 +04:00
|
|
|
void ScoreComponentCollection::PlusEquals(const FeatureFunction* sp, const ScorePair &scorePair)
|
|
|
|
{
|
2013-09-27 12:35:24 +04:00
|
|
|
PlusEquals(sp, scorePair.denseScores);
|
2013-09-08 17:57:31 +04:00
|
|
|
|
2013-09-27 12:35:24 +04:00
|
|
|
std::map<StringPiece, float>::const_iterator iter;
|
|
|
|
for (iter = scorePair.sparseScores.begin(); iter != scorePair.sparseScores.end(); ++iter) {
|
|
|
|
const StringPiece &key = iter->first;
|
|
|
|
float value = iter->second;
|
|
|
|
PlusEquals(sp, key, value);
|
|
|
|
}
|
2013-09-08 17:57:31 +04:00
|
|
|
}
|
|
|
|
|
2008-10-09 03:51:26 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
|