From d8c9fe6f4f42c6494ff3c18719cfd807e1b3ec66 Mon Sep 17 00:00:00 2001
From: Hinome <57831472+RealHinome@users.noreply.github.com>
Date: Thu, 10 Aug 2023 22:23:15 +0200
Subject: [PATCH] Update model

---
 models/html_verificator.py | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/models/html_verificator.py b/models/html_verificator.py
index 33e66aa..b68d975 100644
--- a/models/html_verificator.py
+++ b/models/html_verificator.py
@@ -52,7 +52,7 @@
 
 """Create the text encoder"""
 
-VOCAB_SIZE = 1000
+VOCAB_SIZE = 5000
 encoder = tf.keras.layers.TextVectorization(
     max_tokens=VOCAB_SIZE)
 encoder.adapt(train_dataset.map(lambda text, label: text))
@@ -131,8 +131,3 @@
 
 predictions = new_model.predict(text_text)
 print(predictions)
-
-test_loss, test_acc = new_model.evaluate(test_dataset)
-
-print('Test Loss:', test_loss)
-print('Test Accuracy:', test_acc)
\ No newline at end of file