Python tf.keras.datasets.imdb.get_word_index用法及代碼示例

檢索將單詞映射到 IMDB 數據集中的索引的字典。

用法

tf.keras.datasets.imdb.get_word_index(
    path='imdb_word_index.json'
)

例子：

# Retrieve the training sequences.
(x_train, _), _ = keras.datasets.imdb.load_data()
# Retrieve the word index file mapping words to indices
word_index = keras.datasets.imdb.get_word_index()
# Reverse the word index to obtain a dict mapping indices to words
inverted_word_index = dict((i, word) for (word, i) in word_index.items())
# Decode the first sequence in the dataset
decoded_sequence = " ".join(inverted_word_index[i] for i in x_train[0])