回顧以前的筆記

在keras中，內置了imdb電影評分數據集，來進行評價預測

安裝keras

conda install keras

conda就幫依賴全部搞定，記得加源

導入imdb

from keras.datasets import imdb

數據集簡要說明

• 一個長長的英文句子，有的有幾千單詞，有的有幾十，分類成好的評價和不好的評價

在數據中不是單詞，而是單詞的索引
一共就5萬句子

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

總結

以上是生活随笔為你收集整理的影评情感分类（基于IMDB数据集）的全部內容，希望文章能夠幫你解決所遇到的問題。

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