Rasa_NLU 源碼分析

周末找了個 nlp 相關(guān)的工具,使用起來還不錯照皆,它就是 rasa_nlu, 具有實體識別重绷,意圖分類等功能,在加上一個簡單的意圖操作即可實現(xiàn)簡單的 chatbot 功能纵寝,其類圖如下所示:

Rasa_NLU 類依賴圖

整體程序的入口是在 data_router.py 文件中的 DataRouter 類中论寨,主要作用是將模型以 project 的方式進行管理星立,控制數(shù)據(jù)的流向問題

component_classes 中包含所有 Component 類

# Classes of all known components. If a new component should be added,
# its class name should be listed here.
component_classes = [
    SpacyNLP, MitieNLP,
    SpacyEntityExtractor, MitieEntityExtractor, DucklingExtractor,
    CRFEntityExtractor, DucklingHTTPExtractor,
    EntitySynonymMapper,
    SpacyFeaturizer, MitieFeaturizer, NGramFeaturizer, RegexFeaturizer,
    CountVectorsFeaturizer,
    MitieTokenizer, SpacyTokenizer, WhitespaceTokenizer, JiebaTokenizer,
    SklearnIntentClassifier, MitieIntentClassifier, KeywordIntentClassifier,
    EmbeddingIntentClassifier
]

# Mapping from a components name to its class to allow name based lookup.
registered_components = {c.name: c for c in component_classes}

registered_components 通過將 component_classes 中的類進行迭代并遍歷出名稱 Map

get_component_class 函數(shù)將 名稱轉(zhuǎn)為相應的 Component 類

主要架構(gòu)相關(guān)的文件:

registry.py 文件 主要作用是將 pipeline 中的名稱轉(zhuǎn)為相應的 類爽茴,以及導入相應的模型文件
config.py 配置文件轉(zhuǎn)換
model.py 文件 主要是模型相關(guān)內(nèi)容
類名 說明
RasaNLUModelConfig 用來存放訓練是使用的 pipeline 參數(shù)
Metadata 將 model 目錄下 metadata.json 文件進行解析,并緩存
Trainer 訓練所有相關(guān)的 Component 部分绰垂, 通過 train 函數(shù)進行訓練室奏, 通過 persist 函數(shù)進行持久化存儲
Interpreter 通過訓練好的 pipeline 模型解析文本字符串
Persistor 用于存儲模型在云端 aws,gcs劲装,azure等

在 persist 函數(shù)中胧沫,通過 self.pipeline 緩存內(nèi)容,加上各種參數(shù)以及相應模型文件配置到 metadata.json 文件中

Interpreter 初始化流程

1. 加載 MetaData 數(shù)據(jù)內(nèi)容
2. 根據(jù) metadata.json 中 pipeline 構(gòu)件Component 執(zhí)行序列
3. 初始化 Interpreter 參數(shù)列表

Interpreter 解析文本過程

1. 將文本通過 Message 進行封裝
2. 根據(jù) Component 執(zhí)行序列處理 Message 對象
3. 格式化輸出 Message 對象內(nèi)容

Message 中通過 Map 將所有計算結(jié)果存放在相應的地方最終格式化為輸出結(jié)果

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末占业,一起剝皮案震驚了整個濱河市绒怨,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌谦疾,老刑警劉巖南蹂,帶你破解...
    沈念sama閱讀 216,372評論 6 498
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異念恍,居然都是意外死亡六剥,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,368評論 3 392
  • 文/潘曉璐 我一進店門峰伙,熙熙樓的掌柜王于貴愁眉苦臉地迎上來疗疟,“玉大人,你說我怎么就攤上這事瞳氓〔咄” “怎么了?”我有些...
    開封第一講書人閱讀 162,415評論 0 353
  • 文/不壞的土叔 我叫張陵匣摘,是天一觀的道長店诗。 經(jīng)常有香客問我,道長恋沃,這世上最難降的妖魔是什么必搞? 我笑而不...
    開封第一講書人閱讀 58,157評論 1 292
  • 正文 為了忘掉前任,我火速辦了婚禮囊咏,結(jié)果婚禮上恕洲,老公的妹妹穿的比我還像新娘塔橡。我一直安慰自己,他們只是感情好霜第,可當我...
    茶點故事閱讀 67,171評論 6 388
  • 文/花漫 我一把揭開白布葛家。 她就那樣靜靜地躺著,像睡著了一般泌类。 火紅的嫁衣襯著肌膚如雪癞谒。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,125評論 1 297
  • 那天刃榨,我揣著相機與錄音弹砚,去河邊找鬼。 笑死枢希,一個胖子當著我的面吹牛桌吃,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播苞轿,決...
    沈念sama閱讀 40,028評論 3 417
  • 文/蒼蘭香墨 我猛地睜開眼茅诱,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了搬卒?” 一聲冷哼從身側(cè)響起瑟俭,我...
    開封第一講書人閱讀 38,887評論 0 274
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎契邀,沒想到半個月后摆寄,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,310評論 1 310
  • 正文 獨居荒郊野嶺守林人離奇死亡蹂安,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,533評論 2 332
  • 正文 我和宋清朗相戀三年椭迎,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片田盈。...
    茶點故事閱讀 39,690評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡畜号,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出允瞧,到底是詐尸還是另有隱情简软,我是刑警寧澤,帶...
    沈念sama閱讀 35,411評論 5 343
  • 正文 年R本政府宣布述暂,位于F島的核電站痹升,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏畦韭。R本人自食惡果不足惜疼蛾,卻給世界環(huán)境...
    茶點故事閱讀 41,004評論 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望艺配。 院中可真熱鬧察郁,春花似錦衍慎、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,659評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至麦轰,卻和暖如春乔夯,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背款侵。 一陣腳步聲響...
    開封第一講書人閱讀 32,812評論 1 268
  • 我被黑心中介騙來泰國打工末荐, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人喳坠。 一個月前我還...
    沈念sama閱讀 47,693評論 2 368
  • 正文 我出身青樓鞠评,卻偏偏與公主長得像茂蚓,于是被迫代替她去往敵國和親壕鹉。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 44,577評論 2 353