雾聪
2023-06-28 049e0c35238f32359c2bba8156ffd9e1f033a4f0
add function IsPunc
2个文件已修改
10 ■■■■■ 已修改文件
funasr/runtime/onnxruntime/src/tokenizer.cpp 8 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/runtime/onnxruntime/src/tokenizer.h 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/runtime/onnxruntime/src/tokenizer.cpp
@@ -142,6 +142,14 @@
    return result;
}
bool CTokenizer::IsPunc(string& Punc)
{
    if (m_punc2id.find(Punc) != m_punc2id.end())
        return true;
    else
        return false;
}
vector<string> CTokenizer::SplitChineseString(const string & str_info)
{
    vector<string> list;
funasr/runtime/onnxruntime/src/tokenizer.h
@@ -30,7 +30,7 @@
    vector<string> SplitChineseString(const string& str_info);
    void StrSplit(const string& str, const char split, vector<string>& res);
    void Tokenize(const char* str_info, vector<string>& str_out, vector<int>& id_out);
    bool IsPunc(string& Punc);
};
} // namespace funasr