(self, vec_len, cate2idx)
| 122 | return Entities(ents) |
| 123 | |
| 124 | def vectorize(self, vec_len, cate2idx): |
| 125 | res_vec = np.zeros(vec_len, dtype=int) |
| 126 | for ent in self.ents: |
| 127 | res_vec[ent.start_pos: ent.end_pos] = cate2idx[ent.category] |
| 128 | return res_vec |
| 129 | |
| 130 | def find_entities(self, start_pos, end_pos): |
| 131 | res = [] |