Merge pull request #73 from bigelephant29/user-dict-tag-bug-fix

fix user dict tag bug : wrong buf index assigned
This commit is contained in:
Yanyi Wu 2016-07-21 12:26:16 +08:00 committed by GitHub
commit e3e5f93ca3

View File

@ -4,7 +4,9 @@
#include <iostream> #include <iostream>
#include <fstream> #include <fstream>
#include <map> #include <map>
#include <string>
#include <cstring> #include <cstring>
#include <cstdlib>
#include <stdint.h> #include <stdint.h>
#include <cmath> #include <cmath>
#include <limits> #include <limits>
@ -108,10 +110,17 @@ class DictTrie {
buf.clear(); buf.clear();
Split(line, buf, " "); Split(line, buf, " ");
DictUnit node_info; DictUnit node_info;
MakeNodeInfo(node_info, if(buf.size() == 1){
buf[0], MakeNodeInfo(node_info,
user_word_default_weight_, buf[0],
(buf.size() == 2 ? buf[1] : UNKNOWN_TAG)); user_word_default_weight_,
UNKNOWN_TAG);
} else {
MakeNodeInfo(node_info,
buf[0],
(buf.size() == 2 ? atoi(buf[1].c_str()) : user_word_default_weight_),
(buf.size() == 3 ? buf[2] : buf[1]));
}
static_node_infos_.push_back(node_info); static_node_infos_.push_back(node_info);
if (node_info.word.size() == 1) { if (node_info.word.size() == 1) {
user_dict_single_chinese_word_.insert(node_info.word[0]); user_dict_single_chinese_word_.insert(node_info.word[0]);