MatchZoo深度文本匹配工具

MatchZoo 是中科院開源的一個(gè)文本匹配工具包。 它著重于讓大家更直觀地了解深度文本匹配模型的設(shè)計(jì)、對比和共享。

架構(gòu)

MatchZoo 有三個(gè)主要模塊，分別為數(shù)據(jù)預(yù)處理、模型構(gòu)建 和 訓(xùn)練與評(píng)測。它基于 Keras 開發(fā)，支持 TensorFlow、CNTK 及 Theano，并能在 CPU 與 GPU 上無縫運(yùn)行。

基準(zhǔn)測試

下面，以 WikiQA 數(shù)據(jù)集來舉例說明 MatchZoo 的用法。

以 DRMM 為例，運(yùn)行：

python main.py --phase train --model_file models/wikiqa_config/drmm_wikiqa.config

在測試時(shí)可運(yùn)行：

python main.py --phase predict --model_file models/wikiqa_config/drmm_wikiqa.config

十個(gè)模型的比較結(jié)果如下：

loss 訓(xùn)練曲線圖如下：

MAP 性能測試曲線圖如下：