
MatchZoo 文本匹配工具包
发布日期:2021-05-07 23:34:43
浏览次数:25
分类:精选文章
本文共 1752 字,大约阅读时间需要 5 分钟。
MatchZoo是封装了一系列文本匹配的框架包含以下算法:
官网地址:
Model Detail:
1. DRMM
this model is an implementation of .
- model file: models/drmm.py
- model config: models/drmm_ranking.config
2. MatchPyramid
this model is an implementation of
- model file: models/matchpyramid.py
- model config: models/matchpyramid_ranking.config
3. ARC-I
this model is an implementation of
- model file: models/arci.py
- model config: models/arci_ranking.config
4. DSSM
this model is an implementation of
- model file: models/dssm.py
- model config: models/dssm_ranking.config
5. CDSSM
this model is an implementation of
- model file: models/cdssm.py
- model config: models/cdssm_ranking.config
6. ARC-II
this model is an implementation of
- model file: models/arcii.py
- model config: models/arcii_ranking.config
7. MV-LSTM
this model is an implementation of
- model file: models/mvlstm.py
- model config: models/mvlstm_ranking.config
8. aNMM
this model is an implementation of
- model file: models/anmm.py
- model config: models/anmm_ranking.config
9. DUET
this model is an implementation of
- model file: models/duet.py
- model config: models/duet_ranking.config
10. K-NRM
this model is an implementation of
- model file: models/knrm.py
- model config: models/knrm_ranking.config
11. CONV-KNRM:
this model is an implementation of
- model file: models/convknrm.py
- model config: models/convknrm.config
一些不同模型的测试成绩:
Models | NDCG@3 | NDCG@5 | MAP |
---|---|---|---|
DSSM | 0.5439 | 0.6134 | 0.5647 |
CDSSM | 0.5489 | 0.6084 | 0.5593 |
ARC-I | 0.5680 | 0.6317 | 0.5870 |
ARC-II | 0.5647 | 0.6176 | 0.5845 |
MV-LSTM | 0.5818 | 0.6452 | 0.5988 |
DRMM | 0.6107 | 0.6621 | 0.6195 |
K-NRM | 0.6268 | 0.6693 | 0.6256 |
aNMM | 0.6160 | 0.6696 | 0.6297 |
DUET | 0.6065 | 0.6722 | 0.6301 |
MatchPyramid | 0.6317 | 0.6913 | 0.6434 |
DRMM_TKS | 0.6458 | 0.6956 | 0.6586 |
Environment
- python2.7+
- tensorflow 1.2+
- keras 2.06+
- nltk 3.2.2+
- tqdm 4.19.4+
- h5py 2.7.1+