rename drmmmodel as drmm

NTMC-Community · Dec 14, 2018 · ec8f1b5 · ec8f1b5
1 parent 014daec
commit ec8f1b5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/matchzoo/models/__init__.py b/matchzoo/models/__init__.py
@@ -7,6 +7,7 @@
 from .conv_knrm_model import ConvKNRMModel
 from .duet_model import DUETModel
 from .drmmtks_model import DRMMTKSModel
+from .drmm import DRMM
 
 import matchzoo
 def list_available():

diff --git a/matchzoo/models/drmm_model.py → matchzoo/models/drmm.py b/matchzoo/models/drmm_model.py → matchzoo/models/drmm.py
@@ -1,22 +1,19 @@
 """An implementation of DRMM Model."""
 import typing
-import logging
 
 import keras
 import keras.backend as K
 from keras.activations import softmax
 
 from matchzoo import engine
 
-logger = logging.getLogger(__name__)
 
-
-class DRMMModel(engine.BaseModel):
+class DRMM(engine.BaseModel):
     """
     DRMM Model.
 
     Examples:
-        >>> model = DRMMModel()
+        >>> model = DRMM()
         >>> model.guess_and_fill_missing_params(verbose=0)
         >>> model.build()