Skip to content
This repository has been archived by the owner on Aug 11, 2023. It is now read-only.

Commit

Permalink
Update model
Browse files Browse the repository at this point in the history
  • Loading branch information
@RealHinome
RealHinome authored Aug 10, 2023
1 parent f9ec043 commit d8c9fe6
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions models/html_verificator.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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))
Expand Down Expand Up @@ -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)

0 comments on commit d8c9fe6

Please sign in to comment.