fix GetSegResult

pull/3030/head
TianYuan 3 years ago
parent 5e1ba6d2f3
commit a22a3fe040

@ -213,11 +213,11 @@ int FrontEngineInterface::GenDict(const std::string &dict_file,
}
int FrontEngineInterface::GetSegResult(
std::vector<std::pair<std::string, std::string>> &seg,
std::vector<std::string> &seg_words) {
std::vector<std::pair<std::string, std::string>> *seg,
std::vector<std::string> *seg_words) {
std::vector<std::pair<std::string, std::string>>::iterator iter;
for (iter = seg.begin(); iter != seg.end(); iter++) {
seg_words.push_back((*iter).first);
for (iter = seg->begin(); iter != seg->end(); iter++) {
seg_words->push_back((*iter).first);
}
return 0;
}
@ -724,7 +724,7 @@ int FrontEngineInterface::MergeforModify(
std::vector<std::pair<std::string, std::string>> &seg_word_type,
std::vector<std::pair<std::string, std::string>> &modify_seg_word_type) {
std::vector<std::string> seg_result;
GetSegResult(seg_word_type, seg_result);
GetSegResult(&seg_word_type, &seg_result);
LOG(INFO) << "Before merge, seg result is: "
<< limonp::Join(seg_result.begin(), seg_result.end(), "/");
@ -736,7 +736,7 @@ int FrontEngineInterface::MergeforModify(
modify_seg_word_type = MergeEr(modify_seg_word_type);
seg_result = {};
GetSegResult(modify_seg_word_type, seg_result);
GetSegResult(&modify_seg_word_type, &seg_result);
LOG(INFO) << "After merge, seg result is: "
<< limonp::Join(seg_result.begin(), seg_result.end(), "/");

@ -50,8 +50,8 @@ class FrontEngineInterface : public TextNormalizer {
std::map<std::string, std::string> &map);
// 由 词+词性的分词结果转为仅包含词的结果
int GetSegResult(std::vector<std::pair<std::string, std::string>> &seg,
std::vector<std::string> &seg_words);
int GetSegResult(std::vector<std::pair<std::string, std::string>> *seg,
std::vector<std::string> *seg_words);
// 生成句子的音素音调id。如果音素和音调未分开则 toneids
// 为空fastspeech2反之则不为空(speedyspeech)

Loading…
Cancel
Save