move layers to contrib

tnmurthy · Apr 30, 2018 · aef9789 · aef9789
1 parent 56b076e
commit aef9789
Show file tree

Hide file tree

Showing 9 changed files with 9 additions and 9 deletions.
diff --git a/core_models/memn2n_dialogue/memn2n_weights.npz b/core_models/memn2n_dialogue/memn2n_weights.npz
diff --git a/core_models/reading_comprehension/ngraph_implementation/train.py b/core_models/reading_comprehension/ngraph_implementation/train.py
@@ -20,7 +20,7 @@
 from ngraph.frontends.neon import (Layer, Tanh,
  LSTM, Logistic)
 
-from nlp_architect.layers.match_lstm import (
+from nlp_architect.contrib.ngraph.match_lstm import (
  MatchLSTMCell_withAttention,
  unroll_with_attention,
  AnswerPointer_withAttention,

diff --git a/doc/source/api.rst b/doc/source/api.rst
@@ -65,11 +65,11 @@ These are currently stored in the various models
  nlp_architect.models.chunker_model.TimeDistributedRecurrentOutput
  nlp_architect.models.chunker_model.TimeDistributedRecurrentLast
  nlp_architect.models.chunker_model.TimeDistBiLSTM
- nlp_architect.layers.match_lstm.MatchLSTMCell_withAttention
- nlp_architect.layers.match_lstm.AnswerPointer_withAttention
- nlp_architect.layers.match_lstm.Dropout_Modified
- nlp_architect.layers.match_lstm.LookupTable
- nlp_architect.layers.modified_lookup_table.ModifiedLookupTable
+ nlp_architect.contrib.ngraph.match_lstm.MatchLSTMCell_withAttention
+ nlp_architect.contrib.ngraph.match_lstm.AnswerPointer_withAttention
+ nlp_architect.contrib.ngraph.match_lstm.Dropout_Modified
+ nlp_architect.contrib.ngraph.match_lstm.LookupTable
+ nlp_architect.contrib.ngraph.modified_lookup_table.ModifiedLookupTable
 
 
 

diff --git a/nlp_architect/layers/__init__.py → nlp_architect/contrib/ngraph/__init__.py b/nlp_architect/layers/__init__.py → nlp_architect/contrib/ngraph/__init__.py
diff --git a/nlp_architect/layers/match_lstm.py → nlp_architect/contrib/ngraph/match_lstm.py b/nlp_architect/layers/match_lstm.py → nlp_architect/contrib/ngraph/match_lstm.py
diff --git a/...architect/layers/modified_lookup_table.py → ...t/contrib/ngraph/modified_lookup_table.py b/...architect/layers/modified_lookup_table.py → ...t/contrib/ngraph/modified_lookup_table.py
diff --git a/nlp_architect/models/kvmemn2n.py b/nlp_architect/models/kvmemn2n.py
@@ -22,7 +22,7 @@
 from ngraph.frontends.neon import GaussianInit
 from ngraph.frontends.neon.graph import SubGraph
 from nlp_architect.utils.encodings import position_encoding
-from nlp_architect.layers.modified_lookup_table import ModifiedLookupTable
+from nlp_architect.contrib.ngraph.modified_lookup_table import ModifiedLookupTable
 
 
 class KVMemN2N(Layer):

diff --git a/nlp_architect/models/memn2n_dialogue.py b/nlp_architect/models/memn2n_dialogue.py
@@ -32,7 +32,7 @@
 from ngraph.frontends.neon import Layer
 from ngraph.frontends.neon import GaussianInit
 from ngraph.frontends.neon.graph import SubGraph
-from nlp_architect.layers.modified_lookup_table import ModifiedLookupTable
+from nlp_architect.contrib.ngraph.modified_lookup_table import ModifiedLookupTable
 
 
 class MemN2N_Dialog(Layer):

diff --git a/nlp_architect/utils/encodings.py b/nlp_architect/utils/encodings.py
@@ -19,7 +19,7 @@
 
 def position_encoding(sentence_axis, embedding_axis):
  """
- Position Encoding described in section 4.1 [1]
+ Position Encoding used by the end to end memory network algorithms
  """
  sentence_size = sentence_axis.length
  embedding_size = embedding_axis.length