Generic_DialogSystem/evaluate.py

45 lines
1.3 KiB
Python
Raw Normal View History

2023-05-04 17:30:47 +02:00
import csv
import example_nlu as en
rows = []
def readData():
ii = [5, 6, 7, 8]
jj = [1, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18]
kk = [1, 2]
for i in ii:
for j in jj:
for k in kk:
try:
if j < 10:
with open(f'./data/dialog-0{i}-0{j}-0{k}.tsv', 'r') as read_obj:
csv_reader = csv.reader(read_obj, delimiter='\t')
for row in csv_reader:
if 'user' in row[0]:
rows.append(row[1])
else:
with open(f'./data/dialog-0{i}-{j}-0{k}.tsv', 'r') as read_obj:
csv_reader = csv.reader(read_obj, delimiter='\t')
for row in csv_reader:
if 'user' in row[0]:
rows.append(row[1])
except:
pass
if __name__ == "__main__":
readData()
#l = en.predict('Kup cztery wody')
#print(l)
good = 0
for r in rows:
l = en.predict(str(r))
if l != []:
good += 1
2023-05-04 22:46:28 +02:00
print(f'Accuracy: {good/len(rows)*100}%')
en.predict('rock and stone')