Add validate method to ModelCBOW

sindre0830 · Oct 15, 2023 · 5c09413 · 5c09413
1 parent c2255bd
commit 5c09413
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/source/models.py b/source/models.py
@@ -55,3 +55,7 @@ def forward(self, context_words_idx):
         context_vector = torch.sum(self.input_embeddings(context_words_idx), dim=1)
         output = torch.matmul(context_vector, self.output_embeddings.weight.t())
         return torch.nn.functional.log_softmax(output, dim=1)
+
+    def validate(self, validation_dataloader: datahandler.loaders.ValidationLoader) -> float:
+        validation_dataloader.evaluate_analogies(self.get_embeddings())
+        return validation_dataloader.analogies_accuracy()