| 12345678 |
- def get_chosen_strings(self, v_idx, n_idx):
- v_idx_np = to_np(v_idx)
- n_idx_np = to_np(n_idx)
- res_str = []
- for i in range(n_idx_np.shape[0]):
- v, n = self.verb_map[v_idx_np[i]], self.noun_map[n_idx_np[i]]
- res_str.append(self.word_vocab[v] + " " + self.word_vocab[n])
- return res_str
|
