Update __init__.py

cthoyt · Jun 22, 2019 · cd4b13a · cd4b13a
1 parent e17b420
commit cd4b13a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/nrl/model/__init__.py b/src/nrl/model/__init__.py
@@ -5,11 +5,12 @@
 from .deepwalk import DeepWalkModel
 from .gat2vec import Gat2VecUnsupervisedModel
 from .node2vec import Node2VecModel
-from .util import WalkerModel
+from .util import WalkerModel, Word2VecParameters
 
 __all__ = [
+    'Word2VecParameters',
     'WalkerModel',
     'DeepWalkModel',
     'Gat2VecUnsupervisedModel',
-    'Node2VecModel'
+    'Node2VecModel',
 ]