丁香五月天婷婷久久婷婷色综合91|国产传媒自偷自拍|久久影院亚洲精品|国产欧美VA天堂国产美女自慰视屏|免费黄色av网站|婷婷丁香五月激情四射|日韩AV一区二区中文字幕在线观看|亚洲欧美日本性爱|日日噜噜噜夜夜噜噜噜|中文Av日韩一区二区

您正在使用IE低版瀏覽器,為了您的雷峰網(wǎng)賬號安全和更好的產(chǎn)品體驗,強烈建議使用更快更安全的瀏覽器
此為臨時鏈接,僅用于文章預(yù)覽,將在時失效
人工智能開發(fā)者 正文
發(fā)私信給恒亮
發(fā)送

1

手把手教你如何用 TensorFlow 實現(xiàn)基于 DNN 的文本分類

本文作者: 恒亮 2017-04-19 10:29
導(dǎo)語:怎樣實現(xiàn)文本分類?

手把手教你如何用 TensorFlow 實現(xiàn)基于 DNN 的文本分類

雷鋒網(wǎng)按:本文源自一位數(shù)據(jù)科學(xué)開發(fā)者的個人博客,雷鋒網(wǎng)編譯。

許多開發(fā)者向新手建議:如果你想要入門機器學(xué)習(xí),就必須先了解一些關(guān)鍵算法的工作原理,然后再開始動手實踐。但我不這么認(rèn)為。

我覺得實踐高于理論,新手首先要做的是了解整個模型的工作流程,數(shù)據(jù)大致是怎樣流動的,經(jīng)過了哪些關(guān)鍵的結(jié)點,最后的結(jié)果在哪里獲取,并立即開始動手實踐,構(gòu)建自己的機器學(xué)習(xí)模型。至于算法和函數(shù)內(nèi)部的實現(xiàn)機制,可以等了解整個流程之后,在實踐中進行更深入的學(xué)習(xí)和掌握。

那么問題來了,既然作為初學(xué)者不需要掌握算法細節(jié),但實現(xiàn)模型的過程中又必須用到相關(guān)算法,怎么辦呢?答案是借助于互聯(lián)網(wǎng)上已經(jīng)實現(xiàn)好的函數(shù)庫,例如 TensorFlow。

在本文中,我們將利用 TensorFlow 實現(xiàn)一個基于深度神經(jīng)網(wǎng)絡(luò)(DNN)的文本分類模型,希望對各位初學(xué)者有所幫助。文中所涉完整代碼已經(jīng)在 GitHub 上開源,感興趣的朋友可以在以下鏈接中下載:

http://t.cn/RXiP3Om

下面是正式的教程內(nèi)容:

  關(guān)于 TensorFlow

TensorFlow 是谷歌旗下一個開源的機器學(xué)習(xí)框架。從它的名字就能看出這個框架基本的工作原理:由多維數(shù)組構(gòu)成的張量(tensor)在圖(graph)結(jié)點之間定向流動(flow),從輸入走到輸出。

在 TensorFlow 中,每次運算都可以用數(shù)據(jù)流圖(dataflow graph)的方式表示。每個數(shù)據(jù)流圖都有以下兩個重要元素:

● 一組 tf.Operation 對象,代表運算本身;

● 一組 tf.Tensor 對象,代表被運算的數(shù)據(jù)。

如下圖所示,這里我們以一個簡單的例子說明數(shù)據(jù)流圖具體是怎樣運行的。

手把手教你如何用 TensorFlow 實現(xiàn)基于 DNN 的文本分類

假設(shè)圖中的 x=[1,3,6],y=[1,1,1]。由于 tf.Tensor 被用來表示運算數(shù)據(jù),因此在 TensorFlow 中我們會首先定義兩個 tf.Tensor 常量對象存放數(shù)據(jù)。然后再用 tf.Operation 對象定義圖中的加法運算,具體代碼如下:

import tensorflow as tf

x = tf.constant([1,3,6]) 

y = tf.constant([1,1,1])

op = tf.add(x,y)

現(xiàn)在,我們已經(jīng)定義了數(shù)據(jù)流圖的兩個重要元素:tf.Operation 和 tf.Tensor,那么如何構(gòu)建圖本身呢,具體代碼如下:

import tensorflow as tf

my_graph = tf.Graph()

with my_graph.as_default():

    x = tf.constant([1,3,6]) 

    y = tf.constant([1,1,1])

    op = tf.add(x,y)

至此我們已經(jīng)完成了數(shù)據(jù)流圖的定義,在 TensorFlow 中,只有先定義了圖,才能進行后續(xù)的計算操作(即驅(qū)動數(shù)據(jù)在圖的結(jié)點間定向流動)。這里 TensorFlow 又規(guī)定,要進行后續(xù)的計算,必須通過 tf.Session 來統(tǒng)一管理,因此我們還要定義一個 tf.Session 對象,即會話。

在 TensorFlow 中,tf.Session 專門用來封裝 tf.Operation 在 tf.Tensor 基礎(chǔ)上執(zhí)行的操作環(huán)境。因此,在定義 tf.Session 對象時,也需要傳入相應(yīng)的數(shù)據(jù)流圖(可以通過 graph 參數(shù)傳入),本例中具體的代碼如下:

import tensorflow as tf

my_graph = tf.Graph()

with tf.Session(graph=my_graph) as sess:

    x = tf.constant([1,3,6]) 

    y = tf.constant([1,1,1])

    op = tf.add(x,y)

定義好 tf.Session 之后,我們可以通過 tf.Session.run() 方法來執(zhí)行對應(yīng)的數(shù)據(jù)流圖。run() 方法可以通過 fetches 參數(shù)傳入相應(yīng) tf.Operation 對象,并導(dǎo)入與 tf.Operation 相關(guān)的所有 tf.Tensor 對象,然后遞歸執(zhí)行與當(dāng)前 tf.Operation 有依賴關(guān)系的所有操作。本例中具體執(zhí)行的是求和操作,實現(xiàn)代碼如下:

import tensorflow as tf

my_graph = tf.Graph()

with tf.Session(graph=my_graph) as sess:

    x = tf.constant([1,3,6]) 

    y = tf.constant([1,1,1])

    op = tf.add(x,y)

    result = sess.run(fetches=op)

    print(result)

>>> [2 4 7]

可以看到運算結(jié)果是 [2 4 7]。

  關(guān)于預(yù)測模型

了解 TensorFlow 的基本原理之后,下面的任務(wù)是如何構(gòu)建一個預(yù)測模型。簡單來說,機器學(xué)習(xí)算法 + 數(shù)據(jù)就等于預(yù)測模型。構(gòu)建預(yù)測模型的流程如下圖所示:

手把手教你如何用 TensorFlow 實現(xiàn)基于 DNN 的文本分類

如圖,經(jīng)過數(shù)據(jù)訓(xùn)練的機器學(xué)習(xí)算法就是模型。訓(xùn)練好一個模型之后,輸入待預(yù)測數(shù)據(jù),就能得到相應(yīng)的預(yù)測結(jié)果。大體流程如下圖所示:

手把手教你如何用 TensorFlow 實現(xiàn)基于 DNN 的文本分類

在本例中,我們將要構(gòu)建的模型需要根據(jù)輸入文本,輸出相應(yīng)的類別,即完成文本分類的工作。因此這里的輸入應(yīng)該是文本(text),輸出是類別(category)。更具體地說,本例中我們已經(jīng)事先獲取了標(biāo)記數(shù)據(jù)(即一些已經(jīng)標(biāo)明了類別的文本段),然后用這些數(shù)據(jù)對算法進行訓(xùn)練,最后再用訓(xùn)練好的模型對新文本分類。這一過程也就是通常所說的監(jiān)督學(xué)習(xí)(supervised learning)。另外,由于我們的任務(wù)是對文本數(shù)據(jù)進行分類,所以也屬于分類問題的范疇。

為了構(gòu)建該文本分類模型,下面我們需要介紹一些神經(jīng)網(wǎng)絡(luò)的基礎(chǔ)知識。

  關(guān)于神經(jīng)網(wǎng)絡(luò)

從本質(zhì)上說,神經(jīng)網(wǎng)絡(luò)是計算模型(computational model)的一種。(注:這里所謂計算模型是指通過數(shù)學(xué)語言和數(shù)學(xué)概念描述系統(tǒng)的方法)并且這種計算模型還能夠自動完成學(xué)習(xí)和訓(xùn)練,不需要精確編程。

最原始也是最基礎(chǔ)的一個神經(jīng)網(wǎng)絡(luò)算法模型是感知機模型(Perceptron),關(guān)于感知機模型的詳細介紹請參見這篇博客:

http://t.cn/R5MphRp 

由于神經(jīng)網(wǎng)絡(luò)模型是模擬人類大腦神經(jīng)系統(tǒng)的組織結(jié)構(gòu)而提出的,因此它與人類的腦神經(jīng)網(wǎng)絡(luò)具有相似的結(jié)構(gòu)。

手把手教你如何用 TensorFlow 實現(xiàn)基于 DNN 的文本分類

如上圖所示,一般的神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu)可以分為三層:輸入層、隱蔽層(hidden layer)和輸出層。

為了深入理解神經(jīng)網(wǎng)絡(luò)究竟是如何工作的,我們需要利用 TensorFlow 自己親手構(gòu)建一個神經(jīng)網(wǎng)絡(luò)模型,下面介紹一個具體的實例。(本例部分內(nèi)容源自 GitHub 上的一段開源代碼:鏈接

本例中,我們有兩個隱蔽層(關(guān)于隱蔽層層數(shù)的選擇是另一個問題,更詳細的內(nèi)容可以參考:鏈接)。概括地說,隱蔽層的主要作用是將輸入層的數(shù)據(jù)轉(zhuǎn)換成一種輸出層更便于利用的形式。

手把手教你如何用 TensorFlow 實現(xiàn)基于 DNN 的文本分類

如圖所示,本例中輸入層的每個結(jié)點都代表了輸入文本中的一個詞,接下來是第一個隱蔽層。這里需要注意的是,第一層隱蔽層的結(jié)點個數(shù)選擇也是一項重要的任務(wù),通常被稱為特征選擇。

圖中的每個結(jié)點(也被稱為神經(jīng)元),都會搭配一個權(quán)重。而我們下面所謂訓(xùn)練過程其實就是不斷調(diào)整這些權(quán)重值,讓模型的實際輸出和預(yù)想輸出更匹配的過程。當(dāng)然,除了權(quán)重之外,整個網(wǎng)絡(luò)還要加上一個偏差值。(關(guān)于偏差的詳細介紹詳見:鏈接

對每個結(jié)點做加權(quán)和并加上一個偏差值之后,還需要經(jīng)過激活函數(shù)(activation function)的處理才能輸出到下一層 。實際上,這里激活函數(shù)確定了每個結(jié)點的最終輸出情況,同時為整個模型加入了非線性元素。如果用臺燈來做比喻的話,激活函數(shù)的作用就相當(dāng)于開關(guān)。實際研究中根據(jù)應(yīng)用的具體場景和特點,有各種不同的激活函數(shù)可供選擇,這里屏蔽層選擇的是 ReLu 函數(shù)。

另外圖中還顯示了第二個隱蔽層,它的功能和第一層并沒有本質(zhì)區(qū)別,唯一的不同就是它的輸入是第一層的輸出,而第一層的輸入則是原始數(shù)據(jù)。

最后是輸出層,本例中應(yīng)用了獨熱編碼的方式來對結(jié)果進行分類。這里所謂獨熱編碼是指每個向量中只有一個元素是 1,其他均為 0 的編碼方式。例如我們要將文本數(shù)據(jù)分為三個類別(體育、航空和電腦繪圖),則編碼結(jié)果為:

手把手教你如何用 TensorFlow 實現(xiàn)基于 DNN 的文本分類

這里獨熱編碼的好處是:輸出結(jié)點的個數(shù)恰好等于輸出類別的個數(shù)。此外,輸出層和前面的隱蔽層結(jié)構(gòu)類似,我們也要為每個結(jié)點搭配一個權(quán)重值,加上恰當(dāng)?shù)钠?,最后通過激活函數(shù)的處理。

但本例中輸出層的激活函數(shù)與隱蔽層的激活函數(shù)不同。由于本例的最終目的是輸出每個文本對應(yīng)的類別信息,而這里所有類別之間又是互斥的關(guān)系?;谶@些特點,我們在輸出層選擇了 Softmax 函數(shù)作為激活函數(shù)。該函數(shù)的特點是可以將輸出值轉(zhuǎn)換為 0-1 之間的一個小數(shù)值,并且這些小數(shù)值的和為 1。于是正好可以用這些小數(shù)表示每個類別的可能性分布情況。假如剛才提到的三個類別原本的輸出值為 1.2、0.9 和 0.4,則通過 Softmax 函數(shù)的處理后,得到的結(jié)果為:

手把手教你如何用 TensorFlow 實現(xiàn)基于 DNN 的文本分類

可以看到這三個小數(shù)的和正好為 1。

到目前為止,我們已經(jīng)明確了該神經(jīng)網(wǎng)絡(luò)的數(shù)據(jù)流圖,下面為具體的代碼實現(xiàn):

# Network Parameters
n_hidden_1 = 10        # 1st layer number of features
n_hidden_2 = 5         # 2nd layer number of features
n_input = total_words  # Words in vocab
n_classes = 3          # Categories: graphics, space and baseball

def multilayer_perceptron(input_tensor, weights, biases):
   layer_1_multiplication = tf.matmul(input_tensor, weights['h1'])
   layer_1_addition = tf.add(layer_1_multiplication, biases['b1'])
   layer_1_activation = tf.nn.relu(layer_1_addition)

# Hidden layer with RELU activation
   layer_2_multiplication = tf.matmul(layer_1_activation, weights['h2'])
   layer_2_addition = tf.add(layer_2_multiplication, biases['b2'])
   layer_2_activation = tf.nn.relu(layer_2_addition)

# Output layer with linear activation
   out_layer_multiplication = tf.matmul(layer_2_activation, weights['out'])
   out_layer_addition = out_layer_multiplication + biases['out']

return out_layer_addition

  神經(jīng)網(wǎng)絡(luò)的訓(xùn)練

如前所述,模型訓(xùn)練中一項非常重要的任務(wù)就是調(diào)整結(jié)點的權(quán)重。本節(jié)我們將介紹如何在 TensorFlow 中實現(xiàn)這一過程。

在 TensorFlow 中,結(jié)點權(quán)重和偏差值以變量的形式存儲,即 tf.Variable 對象。在數(shù)據(jù)流圖調(diào)用 run() 函數(shù)的時候,這些值將保持不變。在一般的機器學(xué)習(xí)場景中,權(quán)重值和偏差值的初始取值都通過正太分布確定。具體代碼如下圖所示:

weights = {

    'h1': tf.Variable(tf.random_normal([n_input, n_hidden_1])),

    'h2': tf.Variable(tf.random_normal([n_hidden_1, n_hidden_2])),

    'out': tf.Variable(tf.random_normal([n_hidden_2, n_classes]))

}

biases = {

    'b1': tf.Variable(tf.random_normal([n_hidden_1])),

    'b2': tf.Variable(tf.random_normal([n_hidden_2])),

    'out': tf.Variable(tf.random_normal([n_classes]))

}

以初始值運行神經(jīng)網(wǎng)絡(luò)之后,會得到一個實際輸出值 z,而我們的期望輸出值是 expected,這時我們需要做的就是計算兩者之間的誤差,并通過調(diào)整權(quán)重等參數(shù)使之最小化。一般計算誤差的方法有很多,這里因為我們處理的是分類問題,因此采用交叉熵誤差。(關(guān)于為什么分類問題選用交叉熵,參見:連接

在 TensorFlow 中,我們可以通過調(diào)用 tf.nn.softmax_cross_entropy_with_logits() 函數(shù)來計算交叉熵誤差,因為這里我們的激活函數(shù)選擇了 Softmax ,因此誤差函數(shù)中出現(xiàn)了 softmax_ 前綴。具體代碼如下(代碼中我們同時調(diào)用了 tf.reduced_mean() 函數(shù)來計算平均誤差):

# Construct model

prediction = multilayer_perceptron(input_tensor, weights, biases)

# Define loss

entropy_loss = tf.nn.softmax_cross_entropy_with_logits(logits=prediction, labels=output_tensor)

loss = tf.reduce_mean(entropy_loss)

得到誤差之后,下面的任務(wù)是如何使之最小化。這里我們選擇的方法是最常用的隨機梯度下降法,其直觀的原理圖如下所示:

手把手教你如何用 TensorFlow 實現(xiàn)基于 DNN 的文本分類

同樣,用來計算梯度下降的方法也有很多,這里我們采用了 Adaptive Moment Estimation (Adam) 優(yōu)化法,即自適應(yīng)矩估計的優(yōu)化方法,具體在 TensorFlow 中的體現(xiàn)是 tf.train.AdamOptimizer(learning_rate).minimize(loss) 函數(shù)。這里我們需要傳入 learning_rate 參數(shù)以決定計算梯度時的步進長度。

非常方便的一點是,AdamOptimizer() 函數(shù)封裝了兩種功能:一是計算梯度,二是更新梯度。換句話說,調(diào)用該函數(shù)不但能計算梯度值,還能將計算結(jié)果更新到所有 tf.Variables 對象中,這一點大大降低了編程復(fù)雜度。

具體模型訓(xùn)練部分的代碼如下所示:

learning_rate = 0.001

# Construct model

prediction = multilayer_perceptron(input_tensor, weights, biases)

# Define loss

entropy_loss = tf.nn.softmax_cross_entropy_with_logits(logits=prediction, labels=output_tensor)

loss = tf.reduce_mean(entropy_loss)

optimizer = tf.train.AdamOptimizer(learning_rate=learning_rate).minimize(loss)

  數(shù)據(jù)處理

本例中,我們得到的原始數(shù)據(jù)是許多英文的文本片段,為了將這些數(shù)據(jù)導(dǎo)入模型中,我們需要對原始數(shù)據(jù)進行必要的預(yù)處理過程。這里具體包括兩個部分:

● 為每個單詞編碼;

● 為每個文本片段創(chuàng)建對應(yīng)的張量表示,其中以數(shù)字 1 代表出現(xiàn)了某個單詞,0 表示沒有該單詞。

具體實現(xiàn)代碼如下:

import numpy as np    #numpy is a package for scientific computing

from collections import Counter

vocab = Counter()

text = "Hi from Brazil"

#Get all words

for word in text.split(' '):

    vocab[word]+=1

        

#Convert words to indexes

def get_word_2_index(vocab):

    word2index = {}

    for i,word in enumerate(vocab):

        word2index[word] = i

        

    return word2index

#Now we have an index

word2index = get_word_2_index(vocab)

total_words = len(vocab)

#This is how we create a numpy array (our matrix)

matrix = np.zeros((total_words),dtype=float)

#Now we fill the values

for word in text.split():

    matrix[word2index[word]] += 1

print(matrix)

>>> [ 1.  1.  1.]

從以上代碼可以看到,當(dāng)輸入文本是“Hi from Brazil”時,輸出矩陣是 [ 1. 1. 1.]。而當(dāng)輸入文本只有“Hi”時又會怎么樣呢,具體代碼和結(jié)果如下:

matrix = np.zeros((total_words),dtype=float)

text = "Hi"

for word in text.split():

    matrix[word2index[word.lower()]] += 1

print(matrix)

>>> [ 1.  0.  0.]

可以看到,這時的輸出是 [ 1.  0.  0.]。

相應(yīng)的,我們也可以對類別信息進行編碼,只不過這時使用的是獨熱編碼:

y = np.zeros((3),dtype=float)

if category == 0:

    y[0] = 1.        # [ 1.  0.  0.]

elif category == 1:

    y[1] = 1.        # [ 0.  1.  0.]

else:

     y[2] = 1.       # [ 0.  0.  1.]

  運行模型并預(yù)測

至此我們已經(jīng)對 TensorFlow、神經(jīng)網(wǎng)絡(luò)模型、模型訓(xùn)練和數(shù)據(jù)預(yù)處理等方面有了初步的了解,下面我們將演示如何將這些知識應(yīng)用于實際的數(shù)據(jù)。

這里我們的數(shù)據(jù)來源是 20 Newsgroups,其中包括了 18000 篇新聞稿,覆蓋率 20 個類別,開源免費,下載地址為:

http://t.cn/zY6ssrE 

首先,為了導(dǎo)入這些數(shù)據(jù)集,我們需要借助 scikit-learn 庫。它也是個開源的函數(shù)庫,基于 Python 語言,主要進行機器學(xué)習(xí)相關(guān)的數(shù)據(jù)處理任務(wù)。本例中我們只使用了其中的三個類:comp.graphics,sci.space 和 rec.sport.baseball。

最終數(shù)據(jù)會被分為兩個子集,一個是數(shù)據(jù)訓(xùn)練集,一個是測試集。這里的建議是最好不要提前查看測試數(shù)據(jù)集。因為提前查看測試數(shù)據(jù)會影響我們對模型參數(shù)的選擇,從而影響模型對其他未知數(shù)據(jù)的通用性。

具體的數(shù)據(jù)導(dǎo)入代碼如下:

from sklearn.datasets import fetch_20newsgroups

categories = ["comp.graphics","sci.space","rec.sport.baseball"]

newsgroups_train = fetch_20newsgroups(subset='train', categories=categories)

newsgroups_test = fetch_20newsgroups(subset='test', categories=categories)

在神經(jīng)網(wǎng)絡(luò)術(shù)語中,一個 epoch 過程就是對所有訓(xùn)練數(shù)據(jù)的一個前向傳遞(forward pass)加后向傳遞(backward pass)的完整循環(huán)。這里前向是指根據(jù)現(xiàn)有權(quán)重得到實際輸出值的過程,后向是指根據(jù)誤差結(jié)果反過來調(diào)整權(quán)重的過程。下面我們重點介紹一下 tf.Session.run() 函數(shù),實際上它的完整調(diào)用形式如下:

tf.Session.run(fetches, feed_dict=None, options=None, run_metadata=None)

在文章開頭介紹該函數(shù)時,我們只通過 fetches 參數(shù)傳入了加法操作,但其實它還支持一次傳入多種操作的用法。在面向?qū)嶋H數(shù)據(jù)的模型訓(xùn)練環(huán)節(jié),我們就傳入了兩種操作:一個是誤差計算(即隨機梯度下降),另一個是優(yōu)化函數(shù)(即自適應(yīng)矩估計)。

run() 函數(shù)中另一個重要的參數(shù)是 feed_dict,我們就是通過這個參數(shù)傳入模型每次處理的輸入數(shù)據(jù)。而為了輸入數(shù)據(jù),我們又必須先定義 tf.placeholders。

按照官方文檔的解釋,這里 placeholder 僅僅是一個空客,用于引用即將導(dǎo)入模型的數(shù)據(jù),既不需要初始化,也不存放真實的數(shù)據(jù)。本例中定義 tf.placeholders 的代碼如下:

n_input = total_words # Words in vocab

n_classes = 3         # Categories: graphics, sci.space and baseball

input_tensor = tf.placeholder(tf.float32,[None, n_input],name="input")

output_tensor = tf.placeholder(tf.float32,[None, n_classes],name="output")

在進行實際的模型訓(xùn)練之前,還需要將數(shù)據(jù)分成 batch,即一次計算處理數(shù)據(jù)的量。這時就體現(xiàn)了之前定義 tf.placeholders 的好處,即可以通過 placeholders 定義中的“None”參數(shù)指定一個維度可變的 batch。也就是說,batch 的具體大小可以等后面使用時再確定。這里我們在模型訓(xùn)練階段傳入的 batch 更大,而測試階段可能會做一些改變,因此需要使用可變 batch。隨后在訓(xùn)練中,我們通過 get_batches() 函數(shù)來獲取每次處理的真實文本數(shù)據(jù)。具體模型訓(xùn)練部分的代碼如下:

training_epochs = 10

# Launch the graph

with tf.Session() as sess:

    sess.run(init) #inits the variables (normal distribution, remember?)

    # Training cycle

    for epoch in range(training_epochs):

        avg_cost = 0.

        total_batch = int(len(newsgroups_train.data)/batch_size)

        # Loop over all batches

        for i in range(total_batch):

            batch_x,batch_y = get_batch(newsgroups_train,i,batch_size)

            # Run optimization op (backprop) and cost op (to get loss value)

            c,_ = sess.run([loss,optimizer], feed_dict={input_tensor: batch_x, output_tensor:batch_y})

至此我們已經(jīng)針對實際數(shù)據(jù)完成了模型訓(xùn)練,下面到了應(yīng)用測試數(shù)據(jù)對模型進行測試的時候。在測試過程中,和之前的訓(xùn)練部分類似,我們同樣要定義圖元素,包括操作和數(shù)據(jù)兩類。這里為了計算模型的精度,同時還因為我們對結(jié)果引入了獨熱編碼,因此需要同時得到正確輸出的索引,以及預(yù)測輸出的索引,并檢查它們是否相等,如果不等,要計算相應(yīng)的平均誤差。具體實現(xiàn)代碼和結(jié)果如下:

# Test model

    index_prediction = tf.argmax(prediction, 1)

    index_correct = tf.argmax(output_tensor, 1)

    correct_prediction = tf.equal(index_prediction, index_correct)

    # Calculate accuracy

    accuracy = tf.reduce_mean(tf.cast(correct_prediction, "float"))

    total_test_data = len(newsgroups_test.target)

    batch_x_test,batch_y_test = get_batch(newsgroups_test,0,total_test_data)

    print("Accuracy:", accuracy.eval({input_tensor: batch_x_test, output_tensor: batch_y_test}))

>>> Epoch: 0001 loss= 1133.908114347

    Epoch: 0002 loss= 329.093700409

    Epoch: 0003 loss= 111.876660109

    Epoch: 0004 loss= 72.552971845

    Epoch: 0005 loss= 16.673050320

    Epoch: 0006 loss= 16.481995190

    Epoch: 0007 loss= 4.848220565

    Epoch: 0008 loss= 0.759822878

    Epoch: 0009 loss= 0.000000000

    Epoch: 0010 loss= 0.079848485

    Optimization Finished!

    Accuracy: 0.75

最終可以看到,我們的模型預(yù)測精度達到了 75%,對于初學(xué)者而言,這個成績還是不錯的。至此,我們已經(jīng)通過 TensorFlow 實現(xiàn)了基于神經(jīng)網(wǎng)絡(luò)模型的文本分類任務(wù)。

來源:medium,雷鋒網(wǎng)編譯

雷峰網(wǎng)版權(quán)文章,未經(jīng)授權(quán)禁止轉(zhuǎn)載。詳情見轉(zhuǎn)載須知。

手把手教你如何用 TensorFlow 實現(xiàn)基于 DNN 的文本分類

分享:
相關(guān)文章

編輯

歡迎交流,微信:whl123465
當(dāng)月熱門文章
最新文章
請?zhí)顚懮暾埲速Y料
姓名
電話
郵箱
微信號
作品鏈接
個人簡介
為了您的賬戶安全,請驗證郵箱
您的郵箱還未驗證,完成可獲20積分喲!
請驗證您的郵箱
立即驗證
完善賬號信息
您的賬號已經(jīng)綁定,現(xiàn)在您可以設(shè)置密碼以方便用郵箱登錄
立即設(shè)置 以后再說