Fix and update readme

matin-ghorbani · Mar 30, 2024 · 0c34574 · 0c34574
1 parent a62ec9f
commit 0c34574
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 8 deletions.
diff --git a/session_61_assignment_7.11/README.md b/session_61_assignment_7.11/README.md
@@ -18,11 +18,11 @@
     </tr>    
     </tr>
         <td>100d</td>
-        <td>...</td>
-        <td>...</td>
-        <td>...</td>
-        <td>...</td>
-        <td>...s</td>
+        <td>0.3991</td>
+        <td>0.9470</td>
+        <td>0.4769</td>
+        <td>0.8593</td>
+        <td>0.0993s</td>
     </tr>
     <tr>
         <td>200d</td>

diff --git a/session_61_assignment_7.11/test.py b/session_61_assignment_7.11/test.py
@@ -2,7 +2,6 @@
 from argparse import ArgumentParser, BooleanOptionalAction
 
 from text_classifier import EmojiTextClassifier
-from tqdm import tqdm
 
 args = ArgumentParser()
 args.add_argument('--model', type=str,
@@ -25,7 +24,7 @@
 
 if opt.infer:
     start_time = time()
-    for i in tqdm(range(opt.n_infer)):
+    for i in range(opt.n_infer):
         classifier.predict(opt.sentence)
 
     duration = time() - start_time

diff --git a/session_61_assignment_7.11/text_classifier.py b/session_61_assignment_7.11/text_classifier.py
@@ -36,7 +36,7 @@ def load_feature_vectors(self, file_path: str) -> None:
     def sentence_to_feature_vectors_avg(self, sentence: str) -> np.ndarray[np.floating] | None:
         sentence = sentence.lower()
         words = sentence.strip().split(' ')
-        sum_vectors = np.zeros((50, ))
+        sum_vectors = np.zeros((self.dimension, ))
 
         try:
             for word in words: