Duet tutorial

NTMC-Community · Jan 3, 2019 · 0535b03 · 0535b03
1 parent 6f94af8
commit 0535b03
Show file tree

Hide file tree

Showing 3 changed files with 417 additions and 3 deletions.
diff --git a/matchzoo/models/__init__.py b/matchzoo/models/__init__.py
@@ -6,7 +6,7 @@
 from .arcii_model import ArcIIModel
 from .knrm_model import KNRMModel
 from .conv_knrm_model import ConvKNRMModel
-from .duet_model import DUETModel
+from .duet import DUET
 from .drmmtks_model import DRMMTKSModel
 from .drmm import DRMM
 from .anmm_model import ANMMModel

diff --git a/matchzoo/models/duet_model.py → matchzoo/models/duet.py b/matchzoo/models/duet_model.py → matchzoo/models/duet.py
@@ -8,12 +8,12 @@
 from matchzoo import preprocessors
 
 
-class DUETModel(engine.BaseModel):
+class DUET(engine.BaseModel):
     """
     DUET Model.
 
     Examples:
-        >>> model = DUETModel()
+        >>> model = DUET()
         >>> model.params['embedding_input_dim'] = 1000
         >>> model.params['embedding_output_dim'] = 300
         >>> model.params['lm_filters'] = 32