neuro-lab7/detect.py

36 lines
704 B
Python
Raw Normal View History

2025-12-06 16:36:55 +02:00
from model import m
from preprocessor import frs
import pickle
import time
from sys import exit
from tensorflow.keras.utils import pad_sequences as kps
m.load_weights("model2.keras")
tk = pickle.load(open('tokenizer.pickle', 'rb'))
while True:
t = ""
try:
t = frs(input("Comment: "))
except EOFError:
print("\nExiting")
exit(0)
print(f"Processed: {t}")
s = tk.texts_to_sequences([t])
print(f"Sequence: {s[0]}")
ps = kps(s, maxlen = 100)
p = m.predict(ps)
if p >= 0.75:
print(f"Result: positive ({p[0]})\n")
elif p <= 0.25:
print(f"Result: negative ({p[0]})\n")
else:
print(f"Result: unsure ({p[0]})\n")