Imports for conceptors

nschaetti · Jan 17, 2019 · cf4f5b9 · cf4f5b9
1 parent 9bebeb0
commit cf4f5b9
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/echotorch/nn/__init__.py b/echotorch/nn/__init__.py
@@ -5,6 +5,9 @@
 from .BDESN import BDESN
 from .BDESNPCA import BDESNPCA
 from .BDESNCell import BDESNCell
+from .Conceptor import Conceptor
+from ConceptorNet import ConceptorNet
+from ConceptorNetCell import ConceptorNetCell
 from .ESNCell import ESNCell
 from .ESN import ESN
 from .LiESNCell import LiESNCell
@@ -18,6 +21,6 @@
 from .StackedESN import StackedESN
 
 __all__ = [
-    'BDESN', 'BDESNPCA', 'BDESNCell', 'ESNCell', 'ESN', 'LiESNCell', 'LiESN', 'GatedESN', 'ICACell', 'Identity',
-    'PCACell', 'RRCell', 'SFACell', 'StackedESN'
+    'BDESN', 'BDESNPCA', 'BDESNCell', 'Conceptor', 'ConceptorNet', 'ConceptorNetCell', 'ESNCell', 'ESN', 'LiESNCell',
+    'LiESN', 'GatedESN', 'ICACell', 'Identity', 'PCACell', 'RRCell', 'SFACell', 'StackedESN'
 ]