#347 related files relocation

nicolay-r · Jun 27, 2022 · 9085a82 · 9085a82
1 parent 7eafb4c
commit 9085a82
Show file tree

Hide file tree

Showing 11 changed files with 11 additions and 11 deletions.
diff --git a/arekit/contrib/networks/core/input/embedding/matrix.py b/arekit/contrib/networks/core/input/embedding/matrix.py
@@ -1,8 +1,8 @@
 import logging
 import numpy as np
 
-from arekit.contrib.networks.embeddings.base import Embedding
 from arekit.contrib.networks.core.input.embedding.offsets import TermsEmbeddingOffsets
+from arekit.contrib.networks.embedding import Embedding
 
 logger = logging.getLogger(__name__)
 

diff --git a/arekit/contrib/networks/core/input/embedding/offsets.py b/arekit/contrib/networks/core/input/embedding/offsets.py
@@ -1,5 +1,6 @@
 import logging
-from arekit.contrib.networks.embeddings.base import Embedding
+
+from arekit.contrib.networks.embedding import Embedding
 
 logger = logging.getLogger(__name__)
 

diff --git a/arekit/contrib/networks/embeddings/base.py → arekit/contrib/networks/embedding.py b/arekit/contrib/networks/embeddings/base.py → arekit/contrib/networks/embedding.py
@@ -14,8 +14,6 @@ def __init__(self, matrix, words):
         self.__words = words
         self.__index_by_word = self.__create_index(words)
 
-    # TODO. Add create_term_embedding_func
-
     # region properties
 
     @property

diff --git a/arekit/contrib/networks/embeddings/__init__.py b/arekit/contrib/networks/embeddings/__init__.py
diff --git a/arekit/contrib/networks/handlers/serializer.py b/arekit/contrib/networks/handlers/serializer.py
@@ -11,7 +11,7 @@
 from arekit.contrib.networks.core.input.providers.sample import NetworkSampleRowProvider
 from arekit.contrib.networks.core.input.providers.text import NetworkSingleTextProvider
 from arekit.contrib.networks.core.input.terms_mapping import StringWithEmbeddingNetworkTermMapping
-from arekit.contrib.networks.embeddings.base import Embedding
+from arekit.contrib.networks.embedding import Embedding
 from arekit.contrib.utils.model_io.tf_networks import DefaultNetworkIOUtils
 from arekit.contrib.utils.serializer import InputDataSerializationHelper
 from arekit.contrib.utils.vectorizers.bpe import BPEVectorizer

diff --git a/arekit/contrib/utils/vectorizers/base.py → arekit/contrib/networks/vectorizer.py b/arekit/contrib/utils/vectorizers/base.py → arekit/contrib/networks/vectorizer.py
diff --git a/arekit/contrib/utils/embeddings/rusvectores.py b/arekit/contrib/utils/embeddings/rusvectores.py
@@ -1,8 +1,9 @@
 import numpy as np
 from arekit.common.text.stemmer import Stemmer
-from arekit.contrib.networks.embeddings.base import Embedding
 from gensim.models import KeyedVectors
 
+from arekit.contrib.networks.embedding import Embedding
+
 
 class RusvectoresEmbedding(Embedding):
 

diff --git a/arekit/contrib/utils/embeddings/tokens.py b/arekit/contrib/utils/embeddings/tokens.py
@@ -1,6 +1,6 @@
 import numpy as np
 
-from arekit.contrib.networks.embeddings.base import Embedding
+from arekit.contrib.networks.embedding import Embedding
 from arekit.processing.text.tokens import Tokens
 
 

diff --git a/arekit/contrib/utils/vectorizers/bpe.py b/arekit/contrib/utils/vectorizers/bpe.py
@@ -1,8 +1,8 @@
 import numpy as np
 
 from arekit.common.log_utils import logger
-from arekit.contrib.networks.embeddings.base import Embedding
-from arekit.contrib.utils.vectorizers.base import BaseVectorizer
+from arekit.contrib.networks.embedding import Embedding
+from arekit.contrib.networks.vectorizer import BaseVectorizer
 
 
 class BPEVectorizer(BaseVectorizer):

diff --git a/arekit/contrib/utils/vectorizers/random_norm.py b/arekit/contrib/utils/vectorizers/random_norm.py
@@ -1,6 +1,6 @@
 import numpy as np
 
-from arekit.contrib.utils.vectorizers.base import BaseVectorizer
+from arekit.contrib.networks.vectorizer import BaseVectorizer
 
 
 class RandomNormalVectorizer(BaseVectorizer):

diff --git a/tests/contrib/networks/test_embedding.py b/tests/contrib/networks/test_embedding.py
@@ -1,6 +1,6 @@
 import unittest
 
-from arekit.contrib.networks.embeddings.base import Embedding
+from arekit.contrib.networks.embedding import Embedding
 
 
 class TestModelNamesService(unittest.TestCase):