FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
雾聪
2023-06-28 54931dd4e1a099d7d6f144c4e12e5453deb3aa26 
 funasr/runtime/onnxruntime/src/vocab.h


@@ -7,19 +7,21 @@


#include <vector>


using namespace std;




namespace funasr {


class Vocab {


  private:


    vector<string> vocab;


    bool isChinese(string ch);


    bool isEnglish(string ch);


    void loadVocabFromYaml(const char* filename);


    bool IsChinese(string ch);


    bool IsEnglish(string ch);


    void LoadVocabFromYaml(const char* filename);




  public:


    Vocab(const char *filename);


    ~Vocab();


    int size();


    string vector2string(vector<int> in);


    string vector2stringV2(vector<int> in);


    int Size();


    string Vector2String(vector<int> in);


    string Vector2StringV2(vector<int> in);


};




} // namespace funasr


#endif