change param

HessTaha · May 31, 2020 · 720d9fa · 720d9fa
1 parent c88e0b9
commit 720d9fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -39,7 +39,7 @@ def main(path_to_data: str,
 
     Model = LSTMModel(
         vocab_size=vocab_size,
-        n_classes=2
+        n_classes=n_classes
     )
 
     if n_classes > 2: