From 720d9fa04772715beced1a5b5532103faa3b952d Mon Sep 17 00:00:00 2001
From: HessTaha <tahahessane2@gmail.com>
Date: Sun, 31 May 2020 13:59:04 +0200
Subject: [PATCH] change param

---
 src/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main.py b/src/main.py
index 180ee5e..09a2064 100644
--- 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: