keras基礎實例

電影評價預測

導入數據集

import keras from keras import layers import matplotlib.pyplot as plt %matplotlib inline data = keras.datasets.imdb max_word = 10000 # 加載前10000個單詞 最大不超過10000 (x_train, y_train), (x_test, y_test) = data.load_data(num_words=max_word)

查看數據

x_train.shape, y_train.shape OUT: ((25000,), (25000,)) x_train[0] y_train[0] OUT: 輸出的是 詞匯的index 輸出的是 array([1, 0, 0, ..., 0, 1, 0], dtype=int64) 1 代表 正面評價 0 代表負面怕評價

加載index和詞匯的對應關系

#加載詞匯 word_index = data.get_word_index() #將index和value 互換 index_word = dict((value, key) for key,value in word_index.items())

打印第一個評

總結

以上是生活随笔為你收集整理的keras从入门到放弃(十一）电影评价预测的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。