(self, words, id_list, match_strategy='vote', vote_threshold=0.75, key_weight = {'bow': 1, 'tfidf': 1, 'ngram_tfidf': 1})
| 91 | |
| 92 | |
| 93 | def predict(self, words, id_list, match_strategy='vote', vote_threshold=0.75, key_weight = {'bow': 1, 'tfidf': 1, 'ngram_tfidf': 1}): |
| 94 | res = self._predict( words ) |
| 95 | if match_strategy == 'vote': |
| 96 | a_res_dic = self.vote(res, vote_threshold, key_weight)[1] |
| 97 | return dict(zip( id_list, [a_res_dic[i] for i in id_list] )) |
| 98 | if match_strategy == 'score': |
| 99 | a_res_dic = self.score(res, vote_threshold, key_weight)[1] |
| 100 | return dict(zip( id_list, [a_res_dic[i] for i in id_list] )) |
| 101 | |
| 102 | |
| 103 |