| | |
|---|
| | | #include "vocab.h" |
|---|
| | | #include "yaml-cpp/yaml.h" |
|---|
| | | #include <yaml-cpp/yaml.h> |
|---|
| | | #include <glog/logging.h> |
|---|
| | | |
|---|
| | | #include <fstream> |
|---|
| | | #include <iostream> |
|---|
| | |
|---|
| | | |
|---|
| | | using namespace std; |
|---|
| | | |
|---|
| | | namespace funasr { |
|---|
| | | Vocab::Vocab(const char *filename) |
|---|
| | | { |
|---|
| | | ifstream in(filename); |
|---|
| | |
|---|
| | | YAML::Node config; |
|---|
| | | try{ |
|---|
| | | config = YAML::LoadFile(filename); |
|---|
| | | }catch(...){ |
|---|
| | | printf("error loading file, yaml file error or not exist.\n"); |
|---|
| | | }catch(exception const &e){ |
|---|
| | | LOG(INFO) << "Error loading file, yaml file error or not exist."; |
|---|
| | | exit(-1); |
|---|
| | | } |
|---|
| | | YAML::Node myList = config["token_list"]; |
|---|
| | |
|---|
| | | { |
|---|
| | | return vocab.size(); |
|---|
| | | } |
|---|
| | | |
|---|
| | | } // namespace funasr |
|---|
