Small format fix for word2vec+svm

advaithsrao · Nov 8, 2023 · e6d65ef · e6d65ef
1 parent 473ddd6
commit e6d65ef
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/detector/modeler.py b/detector/modeler.py
@@ -343,7 +343,7 @@ def train(
         X = None
         # Concatenate sparse matrices into a single feature matrix
         for doc in body:
-            doc_vector = self.vectorizer.transform(doc)
+            doc_vector = self.vectorizer.fit_transform(doc)
             if X is None:
                 X = doc_vector
             else:
@@ -373,7 +373,7 @@ def predict(
         X = None
         # Concatenate sparse matrices into a single feature matrix
         for doc in body:
-            doc_vector = self.vectorizer.transform(doc)
+            doc_vector = self.vectorizer.fit_transform(doc)
             if X is None:
                 X = doc_vector
             else: