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