最近看到两种实现,第一种是用循环,第二种很值得借鉴:
# vocab:['xx', 'xx', ...]
word_to_idx = {word: i for i, word in enumerate(vocab)}
idx_to_word = {i: word for i, word in enumerate(vocab)}
最近看到两种实现,第一种是用循环,第二种很值得借鉴:
# vocab:['xx', 'xx', ...]
word_to_idx = {word: i for i, word in enumerate(vocab)}
idx_to_word = {i: word for i, word in enumerate(vocab)}