Update trained model

ingenuity-2-0 · Apr 19, 2022 · c28ba43 · c28ba43
1 parent cac7f0e
commit c28ba43
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git a/chat.py b/chat.py
@@ -38,9 +38,9 @@ def get_response(msg):
     _, predicted = torch.max(output, dim=1)
 
     tag = tags[predicted.item()]
-
     probs = torch.softmax(output, dim=1)
     prob = probs[0][predicted.item()]
+    # print(tag, prob)
     if prob.item() > 0.75:
         for intent in intents['intents']:
             if tag == intent["tag"]:

diff --git a/data_pre.py b/data_pre.py
@@ -56,7 +56,7 @@
 
 
 # Hyper-parameters
-num_epochs = 600
+num_epochs = 1000
 batch_size = 8
 learning_rate = 0.0001
 input_size = len(X_train[0])

diff --git a/datas.pth b/datas.pth