3,jieba gensim 最好別分家之最簡單的類似度實現

簡單的問答已經實現了,那麼問題也跟着出現了,我不能肯定問題必定是"你叫什麼名字",也有多是"你是誰","你叫啥"之類的,這就引出了人工智能中的另外一項技術:算法

天然語言處理(NLP) : 大概意思就是 讓計算機明白一句話要表達的意思,NLP就至關於計算機在思考你說的話,讓計算機知道"你是誰","你叫啥","你叫什麼名字"是一個意思app

這就要作 : 語義類似度機器學習

接下來咱們用Python大法來實現一個簡單的天然語言處理ide

如今又要用到Python強大的三方庫了post

第一個是將中文字符串進行分詞的庫叫 jieba學習

pip install jieba測試

咱們一般把這個庫叫作 結巴分詞 確實是結巴分詞,並且這個詞庫是 made in china , 基本用一下這個結巴分詞:人工智能

import jieba

key_word = "你叫什麼名字"  # 定義一句話,基於這句話進行分詞

cut_word = jieba.cut(key_word)  # 使用結巴分詞中的cut方法對"你叫什麼名字" 進行分詞

print(cut_word)  # <generator object Tokenizer.cut at 0x03676390> 不懂生成器的話,就忽略這裏

cut_word_list = list(cut_word)  # 若是不明白生成器的話,這裏要記得把生成器對象作成列表

print(cut_word_list)  # ['你', '叫', '什麼', '名字']

測試代碼就很明顯了,它很清晰的把我們的中文字符串轉爲列表存儲起來了spa

 

第二個是一個語言訓練庫叫 gensimcode

pip install gensim

這個訓練庫很厲害, 裏面封裝不少機器學習的算法, 是目前人工智能的主流應用庫,這個不是很好理解, 須要必定的Python數據處理的功底

import jieba
import gensim
from gensim import corpora
from gensim import models
from gensim import similarities

l1 = ["你的名字是什麼", "你今年幾歲了", "你有多高你胸多大", "你胸多大"]
a = "你今年多大了"

all_doc_list = []
for doc in l1:
    doc_list = [word for word in jieba.cut(doc)]
    all_doc_list.append(doc_list)

print(all_doc_list)
doc_test_list = [word for word in jieba.cut(a)]

# 製做語料庫
dictionary = corpora.Dictionary(all_doc_list)  # 製做詞袋
# 詞袋的理解
# 詞袋就是將不少不少的詞,進行排列造成一個 詞(key) 與一個 標誌位(value) 的字典
# 例如: {'什麼': 0, '你': 1, '名字': 2, '是': 3, '的': 4, '了': 5, '今年': 6, '幾歲': 7, '多': 8, '有': 9, '胸多大': 10, '高': 11}
# 至於它是作什麼用的,帶着問題往下看

print("token2id", dictionary.token2id)
print("dictionary", dictionary, type(dictionary))

corpus = [dictionary.doc2bow(doc) for doc in all_doc_list]
# 語料庫:
# 這裏是將all_doc_list 中的每個列表中的詞語 與 dictionary 中的Key進行匹配
# 獲得一個匹配後的結果,例如['你', '今年', '幾歲', '了']
# 就能夠獲得 [(1, 1), (5, 1), (6, 1), (7, 1)]
# 1表明的的是 你 1表明出現一次, 5表明的是 了  1表明出現了一次, 以此類推 6 = 今年 , 7 = 幾歲
print("corpus", corpus, type(corpus))

# 將須要尋找類似度的分詞列表 作成 語料庫 doc_test_vec
doc_test_vec = dictionary.doc2bow(doc_test_list)
print("doc_test_vec", doc_test_vec, type(doc_test_vec))

# 將corpus語料庫(初識語料庫) 使用Lsi模型進行訓練
lsi = models.LsiModel(corpus)
# 這裏的只是須要學習Lsi模型來了解的,這裏不作闡述
print("lsi", lsi, type(lsi))
# 語料庫corpus的訓練結果
print("lsi[corpus]", lsi[corpus])
# 得到語料庫doc_test_vec 在 語料庫corpus的訓練結果 中的 向量表示
print("lsi[doc_test_vec]", lsi[doc_test_vec])

# 文本類似度
# 稀疏矩陣類似度 將 主 語料庫corpus的訓練結果 做爲初始值
index = similarities.SparseMatrixSimilarity(lsi[corpus], num_features=len(dictionary.keys()))
print("index", index, type(index))

# 將 語料庫doc_test_vec 在 語料庫corpus的訓練結果 中的 向量表示 與 語料庫corpus的 向量表示 作矩陣類似度計算
sim = index[lsi[doc_test_vec]]

print("sim", sim, type(sim))

# 對下標和類似度結果進行一個排序,拿出類似度最高的結果
# cc = sorted(enumerate(sim), key=lambda item: item[1],reverse=True)
cc = sorted(enumerate(sim), key=lambda item: -item[1])
print(cc)

text = l1[cc[0][0]]

print(a,text)
前方高能
相關文章
相關標籤/搜索