mosesdecoder/mert/Optimizer.h

123 lines
3.3 KiB
C
Raw Normal View History

#ifndef MERT_OPTIMIZER_H_
#define MERT_OPTIMIZER_H_
#include <vector>
#include <string>
2012-02-08 21:11:56 +04:00
#include "Data.h"
#include "FeatureData.h"
#include "Scorer.h"
#include "Types.h"
static const float kMaxFloat = std::numeric_limits<float>::max();
namespace MosesTuning
{
2013-05-29 21:16:15 +04:00
class Point;
/**
* Abstract optimizer class.
*/
class Optimizer
{
protected:
Scorer *m_scorer; // no accessor for them only child can use them
FeatureDataHandle m_feature_data; // no accessor for them only child can use them
unsigned int m_num_random_directions;
2011-11-11 15:40:59 +04:00
const std::vector<bool>& m_positive;
public:
Optimizer(unsigned Pd, const std::vector<unsigned>& i2O, const std::vector<bool>& positive, const std::vector<parameter_t>& start, unsigned int nrandom);
2013-05-29 21:16:15 +04:00
void SetScorer(Scorer *scorer) {
m_scorer = scorer;
}
void SetFeatureData(FeatureDataHandle feature_data) {
m_feature_data = feature_data;
}
virtual ~Optimizer();
unsigned size() const {
return m_feature_data ? m_feature_data->size() : 0;
}
/**
* Generic wrapper around TrueRun to check a few things. Non virtual.
*/
statscore_t Run(Point&) const;
/**
* Main function that performs an optimization.
*/
virtual statscore_t TrueRun(Point&) const = 0;
/**
* Given a set of lambdas, get the nbest for each sentence.
*/
void Get1bests(const Point& param,std::vector<unsigned>& bests) const;
/**
* Given a set of nbests, get the Statistical score.
*/
statscore_t GetStatScore(const std::vector<unsigned>& nbests) const {
return m_scorer->score(nbests);
}
statscore_t GetStatScore(const Point& param) const;
std::vector<statscore_t> GetIncStatScore(const std::vector<unsigned>& ref, const std::vector<std::vector<std::pair<unsigned,unsigned> > >& diffs) const;
/**
* Get the optimal Lambda and the best score in a particular direction from a given Point.
*/
statscore_t LineOptimize(const Point& start, const Point& direction, Point& best) const;
};
/**
* Default basic optimizer.
* This class implements Powell's method.
*/
class SimpleOptimizer : public Optimizer
{
private:
const float kEPS;
public:
SimpleOptimizer(unsigned dim, const std::vector<unsigned>& i2O, const std::vector<bool>& positive,
const std::vector<parameter_t>& start, unsigned int nrandom)
: Optimizer(dim, i2O, positive, start,nrandom), kEPS(0.0001f) {}
virtual statscore_t TrueRun(Point&) const;
};
/**
* An optimizer with random directions.
*/
class RandomDirectionOptimizer : public Optimizer
{
private:
const float kEPS;
public:
RandomDirectionOptimizer(unsigned dim, const std::vector<unsigned>& i2O, const std::vector<bool>& positive,
const std::vector<parameter_t>& start, unsigned int nrandom)
2013-05-29 21:16:15 +04:00
: Optimizer(dim, i2O, positive, start, nrandom), kEPS(0.0001f) {}
virtual statscore_t TrueRun(Point&) const;
};
/**
* Dumb baseline optimizer: just picks a random point and quits.
*/
class RandomOptimizer : public Optimizer
{
public:
RandomOptimizer(unsigned dim, const std::vector<unsigned>& i2O, const std::vector<bool>& positive,
const std::vector<parameter_t>& start, unsigned int nrandom)
2013-05-29 21:16:15 +04:00
: Optimizer(dim, i2O, positive, start, nrandom) {}
virtual statscore_t TrueRun(Point&) const;
};
}
#endif // OPTIMIZER_H