modify keyword_demo

This commit is contained in:
wyy 2014-04-20 00:23:42 +08:00
parent ae3e0a1b6a
commit be3773920a

View File

@ -6,8 +6,11 @@ int main(int argc, char ** argv)
KeywordExtractor extractor("../dict/jieba.dict.utf8", "../dict/hmm_model.utf8", "../dict/idf.utf8", "../dict/stop_words.utf8");
string s("我是拖拉机学院手扶拖拉机专业的。不用多久我就会升职加薪当上CEO走上人生巅峰。");
vector<pair<string, double> > wordweights;
vector<string> words;
size_t topN = 5;
extractor.extract(s, wordweights, topN);
cout<< s << '\n' << wordweights << endl;
extractor.extract(s, words, topN);
cout<< s << '\n' << words << endl;
return EXIT_SUCCESS;
}