std::vector<std::string> recognize(const std::vector<std::string>& words, const std::vector<std::string>& postags) { std::vector<std::string> netags; if (model == NULL) { std::cerr << "NER: Model not loaded!" << std::endl; } else { ner_recognize(model, words, postags, netags); } return netags; }
// perform ner over xml int LTP::ner(XML4NLP & xml) { if ( xml.QueryNote(NOTE_NE) ) { return 0; } // dependency int ret = postag(xml); if (0 != ret) { ERROR_LOG("in LTP::ner, failed to perform postag preprocess"); return ret; } void * ner = _resource.GetNER(); if (NULL == ner) { ERROR_LOG("in LTP::ner, failed to init a ner."); return kNERError; } int stnsNum = xml.CountSentenceInDocument(); if (stnsNum == 0) { ERROR_LOG("in LTP::ner, number of sentence equals 0"); return kEmptyStringError; } for (int i = 0; i < stnsNum; ++ i) { vector<string> vecWord; vector<string> vecPOS; vector<string> vecNETag; if (xml.GetWordsFromSentence(vecWord, i) != 0) { ERROR_LOG("in LTP::ner, failed to get words from xml"); return kReadXmlError; } if (xml.GetPOSsFromSentence(vecPOS, i) != 0) { ERROR_LOG("in LTP::ner, failed to get postags from xml"); return kNERError; } if (0 == vecWord.size()) { ERROR_LOG("Input sentence is empty."); return kEmptyStringError; } if (vecWord.size() > MAX_WORDS_NUM) { ERROR_LOG("Input sentence is too long."); return kSentenceTooLongError; } if (0 == ner_recognize(ner, vecWord, vecPOS, vecNETag)) { ERROR_LOG("in LTP::ner, failed to perform ner on sent. #%d", i+1); return kNERError; } xml.SetNEsToSentence(vecNETag, i); } xml.SetNote(NOTE_NE); return 0; }