【自然語言處理 — 概念篇】 來認識一下詞向量(Word Embedding or Word Vector)吧

安裝套件

• gensim: 文檔相似度計算和詞向量表示等自然語言處理(NLP)任務的Python庫。
• jieba: 將中文語句切成最小單位的詞。
• matplotlib: 繪製向量圖表。

!pip install gensim



!pip install jieba



!pip install matplotlib

定義中文語句並進行斷詞

這邊我們就設計個4句的範本進行後續的向量處理。

由於NLP世界中最小的單位是「詞」, 因此我們就要藉由jieba這套斷詞工具幫我們預先進行斷詞。

import jieba

sentences = [

    list(jieba.cut('我喜歡吃蘋果')),

    list(jieba.cut('蘋果很好吃')),

    list(jieba.cut('水果是健康的')),

    list(jieba.cut('梨子也很好吃')),

    list(jieba.cut('我也喜歡吃柳丁')),

    list(jieba.cut('蘋果柳丁都是一種水果')),

    list(jieba.cut('蘋果是一種又香又甜的水果')),

    list(jieba.cut('梨子跟柳丁也是一種又香又甜的水果')),

]



sentences

[['我', '喜歡', '吃', '蘋果'],

 ['蘋果', '很', '好吃'],

 ['水果', '是', '健康', '的'],

 ['梨子', '也', '很', '好吃'],

 ['我', '也', '喜歡', '吃', '柳丁'],

 ['蘋果', '柳丁', '都', '是', '一種', '水果'],

 ['蘋果', '是', '一種', '又', '香', '又', '甜', '的', '水果'],

 ['梨子', '跟', '柳丁', '也', '是', '一種', '又', '香', '又', '甜', '的', '水果']]

訓練詞向量模型並取得詞向量

所謂的「詞向量」也稱為詞嵌入(Word embeddings), 當我們處理文本時，詞向量就是一種將詞語轉換為數字表示的方法。這些數字表示的向量具有特定的性質，能夠表示詞語之間的語義和關聯性。

總而言之，詞向量是一種將詞語轉換為數字向量表示的方法，能夠捕捉詞語之間的語義相似性，並在文本處理任務中發揮重要作用。

底下會得到一堆向量數值，嗯..., 看不懂, 怎麼辦？ 沒關係, 圖表是我們最大的幫手, 下一步我們就將這些向量值跟詞進行圖表的繪製吧！

from gensim.models import Word2Vec



# 訓練詞向量模型

model = Word2Vec(sentences, window=5, min_count=1, workers=4)



# 獲取所有詞

vocab = model.wv.index_to_key



# 獲取所有詞向量

vectors = model.wv[vocab]



vectors

array([[-5.3630857e-04,  2.3646720e-04,  5.1041231e-03, ...,

        -7.0426278e-03,  9.0159255e-04,  6.3935029e-03],

       [-8.6233551e-03,  3.6638153e-03,  5.1871063e-03, ...,

        -2.3919728e-03, -9.5016956e-03,  4.5148465e-03],

       [ 8.8638728e-05,  3.0799997e-03, -6.8148547e-03, ...,

         5.1378825e-04,  8.2169194e-03, -7.0118154e-03],

       ...,

       [ 1.3001668e-03, -9.8043028e-03,  4.5877625e-03, ...,

        -2.5875687e-03, -7.7571664e-03,  4.1926014e-03],

       [ 1.8002307e-03,  7.0460914e-03,  2.9446983e-03, ...,

        -1.8595541e-03,  3.6117458e-03, -7.0364270e-03],

       [ 9.7315907e-03, -9.7780135e-03, -6.4999200e-03, ...,

        -2.7000697e-03,  3.8249956e-03,  3.3858197e-04]], dtype=float32)

下載中文字型讓圖表可以顯示中文

請參考「🖋 【Google Colab Python系列】 視覺化資料Matplotlib 如何繪製出中文？」

import matplotlib as mpl

import matplotlib.font_manager as fm

import matplotlib.pyplot as plt



# 下載繁體中文字型

!wget -O SourceHanSerifTW-VF.ttf https://github.com/adobe-fonts/source-han-serif/raw/release/Variable/TTF/Subset/SourceHanSerifTW-VF.ttf



# 加入字型檔

fm.fontManager.addfont('SourceHanSerifTW-VF.ttf')



# 設定字型

# 

mpl.rc('font', family='Source Han Serif TW VF')

針對訓練完成的模型繪製二維向量圖

在前面我們已經展示了word2vec的訓練方法, 訓練完畢之後我們應該要如何觀察這些詞向量分佈位置是不是我們所預期的, 但礙於高維度對我們人類來說難以理解，因此可以透過降維的方式繪製。

t-SNE(t-Distributed Stochastic Neighbor Embedding)是一種降維算法, 用於將高維數據映射到低維空間, 通常用於視覺化高維數據。

可以看到繪圖後的結果, 本應該相似的蘋果、梨子...等水果應該距離要相近的, 但我們的訓練樣本數實在太少了, 這樣的分佈狀況會有點失真， 這邊只是為了以圖來呈現我們的詞向量, 真正可用的模型是需要大量資料來進行訓練的, 以下我們也會使用已訓練完成的基礎模型來進行相似詞的搜尋。

import matplotlib.pyplot as plt





# 將詞向量降維為2D

from sklearn.manifold import TSNE

# 使用 t-SNE 進行詞向量降維

# perplexity 是用於控制降維後的分佈緊密度的參數。

tsne = TSNE(n_components=2, perplexity=10)

vectors_2d = tsne.fit_transform(vectors)



# 繪製散點圖

plt.scatter(vectors_2d[:, 0], vectors_2d[:, 1])



# 將詞語標籤添加到對應的點上

for i, word in enumerate(vocab):

    plt.annotate(word, xy=(vectors_2d[i, 0], vectors_2d[i, 1]), fontsize=8)





# 顯示圖形

plt.show()

下載預訓練模型

上面我可以看到由於資料量不足, 導致訓練出來的模型沒有參考價值, 因此我們就使用別人訓練好的中文模型來進行接下來下的實驗吧。

這次使用的模型是fasttext的wiki.zh.vec, 如果有其他語言需求的朋友可以到這裡進行下載：

{% embed url="https://fasttext.cc/docs/en/pretrained-vectors.html" %}

import urllib.request



url = 'https://dl.fbaipublicfiles.com/fasttext/vectors-wiki/wiki.zh.vec'



urllib.request.urlretrieve(url, './wiki.zh.vec')

載入預訓練模型

載入模型之後, 我們可以對某個詞取出向量值。

from gensim.models import KeyedVectors



# 載入中文詞向量模型

vector_file_path = './wiki.zh.vec'

wiki_model = KeyedVectors.load_word2vec_format(vector_file_path, binary=False)



# 使用詞向量模型進行相關操作

# 例如，取得詞向量

wiki_vec = wiki_model['詞']



wiki_vec

找出「詞」的相似詞並以圖來分析

1. 找相似詞。
2. 降維度。
3. 縮放向量值: 1 ~ -1之間。
4. 設定門檻值為1.0, 並且將距離在1.0以內的相似詞連接起來。
5. 繪製圖表進行分析。

import matplotlib.pyplot as plt

import numpy as np

from sklearn.manifold import TSNE

from sklearn.preprocessing import MinMaxScaler



input_word = '積極'



# 輸入詞的相似詞及向量資訊

similar_words = wiki_model.most_similar(input_word, topn=20)

similar_words, scores = zip(*similar_words)

similar_vectors = [wiki_model[x] for x in similar_words]



# 將「輸入詞」與「相似詞」總和在一起, 統一進行以下的降維, 並將「輸入詞」擺放於第一位以利識別

words = [input_word] + list(similar_words)

input_vector = wiki_model[input_word]

vectors = np.vstack(([input_vector], similar_vectors))





# 使用 t-SNE 降維

tsne = TSNE(n_components=2, perplexity=10)

vectors_2d = tsne.fit_transform(vectors)



# 縮放向量值到範圍 [-1, 1]

scaler = MinMaxScaler(feature_range=(-1, 1))

vectors_2d = scaler.fit_transform(vectors_2d)



# 繪製散點圖

plt.figure(figsize=(10, 6))



# 繪製輸入詞本身的點

plt.scatter(vectors_2d[0, 0], vectors_2d[0, 1], color='red', label=input_word)



# 將輸入詞標籤添加到對應的點上

plt.annotate(input_word, xy=(vectors_2d[0, 0], vectors_2d[0, 1]), fontsize=16)



# 繪製相似詞的點

plt.scatter(vectors_2d[1:, 0], vectors_2d[1:, 1])



# 將相似詞標籤添加到對應的點上

for i, x in enumerate(similar_words[1:]):

    plt.annotate(x, xy=(vectors_2d[i+1, 0], vectors_2d[i+1, 1]), fontsize=16)



threshold = 1.0

# 繪製連線

for i in range(1, len(vectors_2d)):

    distance = np.linalg.norm(vectors_2d[0, :] - vectors_2d[i, :])

    if distance <= threshold:

      plt.plot([vectors_2d[0, 0], vectors_2d[i, 0]], [vectors_2d[0, 1], vectors_2d[i, 1]], color='gray', linestyle='dotted')

      midpoint = (vectors_2d[0, :] + vectors_2d[i, :]) / 2

      plt.annotate(f'{distance:.2f}', xy=midpoint, ha='center', va='center')



plt.xlabel('x')

plt.ylabel('y')

plt.title(f'詞 "{input_word}" 的相似詞')

plt.legend()

plt.tight_layout()

plt.show()

今天的範例都在這裡「📦 wordvector.ipynb」歡迎自行取用。

如何使用請參閱「【Google Colab系列】台股分析預備式： Colab平台與Python如何擦出火花？」。

結語

將向量透過圖表化的方式可以幫助我們更容易了解到NLP的基礎處理方式, 一步步拆解之後, 我們就不會對這種技術感到害怕了, 整個操作下來感覺科學家的智慧真的是令人讚嘆, 能夠將我們人類的語句想辦法讓機器可以理解甚至進行推理, 就以情緒分類來說, 訓練的過程中考慮了詞彙在上下文中的出現頻率和分佈情況, 並且在詞向量空間中可能會形成特定的集群或分布, 透過這樣的觀察就能夠推測出可能的情緒。

------------------------------------------------------------------------------------------------

喜歡撰寫文章的你，不妨來了解一下：

Web3.0時代下為創作者、閱讀者打造的專屬共贏平台 - 為什麼要加入？

歡迎加入一起練習寫作，賺取知識！