mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-12-25 21:03:22 +03:00
70 lines
1.7 KiB
C++
70 lines
1.7 KiB
C++
/*
|
|
* MiraFeatureVector.h
|
|
* kbmira - k-best Batch MIRA
|
|
*
|
|
* An alternative to the existing SparseVector
|
|
* and FeatureDataItem combo. Should be as memory
|
|
* efficient, and a little more time efficient,
|
|
* and should save me from constantly hacking
|
|
* SparseVector
|
|
*/
|
|
|
|
#ifndef MERT_MIRA_FEATURE_VECTOR_H
|
|
#define MERT_MIRA_FEATURE_VECTOR_H
|
|
|
|
#include <vector>
|
|
#include <iostream>
|
|
|
|
#include "FeatureDataIterator.h"
|
|
|
|
namespace MosesTuning
|
|
{
|
|
|
|
|
|
typedef FeatureStatsType ValType;
|
|
|
|
class MiraFeatureVector
|
|
{
|
|
public:
|
|
MiraFeatureVector() {}
|
|
MiraFeatureVector(const FeatureDataItem& vec);
|
|
//Assumes that features in sparse with id < num_dense are dense features
|
|
MiraFeatureVector(const SparseVector& sparse, size_t num_dense);
|
|
MiraFeatureVector(const MiraFeatureVector& other);
|
|
MiraFeatureVector(const std::vector<ValType>& dense,
|
|
const std::vector<std::size_t>& sparseFeats,
|
|
const std::vector<ValType>& sparseVals);
|
|
|
|
ValType val(std::size_t index) const;
|
|
std::size_t feat(std::size_t index) const;
|
|
std::size_t size() const;
|
|
ValType sqrNorm() const;
|
|
|
|
friend MiraFeatureVector operator-(const MiraFeatureVector& a,
|
|
const MiraFeatureVector& b);
|
|
|
|
friend std::ostream& operator<<(std::ostream& o, const MiraFeatureVector& e);
|
|
|
|
friend bool operator==(const MiraFeatureVector& a,const MiraFeatureVector& b);
|
|
|
|
private:
|
|
//Ignore any sparse features with id < ignoreLimit
|
|
void InitSparse(const SparseVector& sparse, size_t ignoreLimit = 0);
|
|
|
|
std::vector<ValType> m_dense;
|
|
std::vector<std::size_t> m_sparseFeats;
|
|
std::vector<ValType> m_sparseVals;
|
|
};
|
|
|
|
} // namespace
|
|
|
|
#endif // MERT_FEATURE_VECTOR_H
|
|
|
|
// --Emacs trickery--
|
|
// Local Variables:
|
|
// mode:c++
|
|
// c-basic-offset:2
|
|
// End:
|
|
|
|
|