Rename models.CBOW to ModelCBOW class

sindre0830 · Oct 15, 2023 · 231387b · 231387b
1 parent c02cecb
commit 231387b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/source/architechtures/cbow.py b/source/architechtures/cbow.py
@@ -34,7 +34,7 @@ def run() -> None:
     validation_dataloader.build(vocabulary)
     # fit model and get embeddings
     utils.print_divider()
-    model = models.CBOW(config.device, config.vocabulary_size, config.embedding_size, vocabulary.padding_index)
+    model = models.ModelCBOW(config.device, config.vocabulary_size, config.embedding_size, vocabulary.padding_index)
     embeddings = model.get_embeddings()
     utils.print_divider()
     # evaluate embeddings

diff --git a/source/models.py b/source/models.py
@@ -4,7 +4,7 @@
 import numpy as np
 
 
-class CBOW(torch.nn.Module):
+class ModelCBOW(torch.nn.Module):
     def __init__(
             self,
             device: str,