keras基礎(chǔ)實(shí)例

電影評價(jià)預(yù)測

導(dǎo)入數(shù)據(jù)集

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

查看數(shù)據(jù)

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 代表 正面評價(jià) 0 代表負(fù)面怕評價(jià)

加載index和詞匯的對應(yīng)關(guān)系

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

打印第一個(gè)評

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。