third_codes/cppjieba
1
0
Fork 0
mirror of https://github.com/yanyiwu/cppjieba.git synced 2025-07-18 00:00:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

user dict support user weight and user tag

Browse Source
This commit is contained in:
bigelephant29 2016-07-21 10:38:46 +08:00
parent b82acaf71e
commit 2e1b6e0443
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
include/cppjieba/DictTrie.hpp
View File

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