add_tone_seq2seq/accuracy.py
2019-12-27 17:19:22 +07:00

42 lines
1.1 KiB
Python

import requests
import json
# with open('data/labels3_test.txt') as f:
# lines = f.readlines()
with open('data/labels_vinh2.txt') as f:
lines = f.readlines()
def preprocessing(text):
text = text.lower().replace('\n', '').strip()
return text
count = 0
count_correction = 0
for line in lines:
ls_item = line.split(', ')
# print(ls_item[0], ls_item[1])
# rs = requests.get('http://127.0.0.1:5000/spell/' + preprocessing(ls_item[0]))
# a = json.loads(rs.text)
# # print(a['result'])
# print(a['result'], preprocessing(ls_item[1]))
if preprocessing(ls_item[0]) != preprocessing(ls_item[1]):
count += 1
rs = requests.get('http://127.0.0.1:8000/spell/' + preprocessing(ls_item[0]))
a = json.loads(rs.text)
if preprocessing(ls_item[1]) == a['result']:
count_correction += 1
else:
# print(a['result'], preprocessing(ls_item[1]))
pass
else:
print(preprocessing(ls_item[0]))
print(len(lines))
print(count)
print(count_correction)
rs = requests.get('http://127.0.0.1:8000/spell/đương nhiển rồi')
a = json.loads(rs.text)
print(a['result'])