cppjieba/src/HMMSegment.h
2013-08-25 21:31:18 +08:00

53 lines
1.2 KiB
C++

#ifndef CPPJIBEA_HMMSEGMENT_H
#define CPPJIBEA_HMMSEGMENT_H
#include <iostream>
#include <fstream>
#include <memory.h>
#include "cppcommon/headers.h"
#include "globals.h"
#include "TransCode.h"
namespace CppJieba
{
using namespace CPPCOMMON;
class HMMSegment
{
private:
/*
* STATUS:
* 0:B, 1:E, 2:M, 3:S
* */
enum {B = 0, E = 1, M = 2, S = 3, STATUS_SUM = 4};
char _statMap[STATUS_SUM];
double _startProb[STATUS_SUM];
double _transProb[STATUS_SUM][STATUS_SUM];
EmitProbMap _emitProbB;
EmitProbMap _emitProbE;
EmitProbMap _emitProbM;
EmitProbMap _emitProbS;
vector<EmitProbMap* > _emitProbVec;
public:
HMMSegment();
~HMMSegment();
public:
bool init();
bool dispose();
public:
bool loadModel(const char* const filePath);
bool cut(const string& str, vector<string>& res);
bool viterbi(const vector<uint16_t>& unico, vector<uint>& status);
private:
bool _getLine(ifstream& ifile, string& line);
bool _loadEmitProb(const string& line, EmitProbMap& mp);
bool _decodeOne(const string& str, uint16_t& res);
double _getEmitProb(const EmitProbMap& mp, uint16_t key, double defVal);
double _getEmitProb(const EmitProbMap* ptMp, uint16_t key, double defVal);
};
}
#endif