Update test data

This commit is contained in:
Aleksey Filippov 2023-06-03 00:19:03 +04:00
parent 3eb01b064c
commit ef33282cbb

92
main.py
View File

@ -3,61 +3,63 @@
import sys import sys
from src.myontology import MyOntology from src.myontology import MyOntology
from src.nlp import NLP
from src.parse_tree.parse_tree import ParseTree
from src.syntax import Syntax
data = [ test_data = [
['получить кредит обучение'], 'как получить кредит на обучение',
['получить кредит образование'], 'Как получить кредит на образование?',
['купить товар рассрочка'], 'Можно ли купить товар в рассрочку?',
['кредит ставка ипотека'], 'какая ставка кредита на ипотеку?',
['самая низкий ставка кредиту'], 'Самая низкая ставка по кредиту?',
['оформить кредит'], 'Как оформить кредит?',
['ставка кредит'], 'Ставка кредита',
['ипотека кредит ставка', 'какой ставка'], 'Какая ставка кредита на ипотеку?',
['получить деньга карта', 'заблокировать'], 'Как получить деньги с заблокированной карты?',
['снять деньга карта'], 'Как снять деньги с заблокированной карты',
['какой баланс карта'], 'Какой баланс карты',
['лимит карта'], 'Какой лимит карты',
['проходить оплата', 'быть деньга'], 'Почему не проходит оплата, хотя деньги есть',
['деньга карта'], 'Сколько денег на карте',
['деньга счёт'], 'Сколько денег на счете',
['разблокировать карта'], 'Как разблокировать карту?',
['заблокировать карта'], 'Как заблокировать карту?',
['какой условие ипотека'], 'Какие условия по ипотеке?',
['какой условие кредит жильё'], 'Какие условия кредита на жилье',
['какой срок кредит жильё'], 'Какой срок кредита на жилье',
['какой срок ипотека'], 'Какой срок ипотеки',
['потерять карта'], 'Я потеряла карту. Что делать?',
['застрять банкомат'], 'Карта застряла в банкомате. Что делать?',
['украсть карта'], 'У меня украли карту. Что делать?',
['какой остаток карта'], 'Какой остаток по карте?',
['купить дом'], 'Хочу купить дом',
['купить дом'], 'Хочу купить дом . Что делать?',
['купить товар кредит'], 'Можно ли купить товар в кредит?',
['срок ипотека'], 'Какой срок по ипотеке',
['купить дом кредит'], 'Хочу купить дом в кредит Что делать?',
['снизить ставка кредиту'], 'Можно ли снизить ставку по кредиту?',
['рефинансировать кредит'], 'Можно ли рефинансировать кредит?',
['снизить ставка кредиту'], 'Можно ли снизить ставку по кредиту другого банка?',
['ли друг банка банк, перевести банк'], 'У меня кредит в другом банке. Можно ли его перевести в ваш банк?',
['провести рефинансирование кредит'], 'Можно провести рефинансирование кредита?',
['рефинансировать ипотека'] 'Можно ли рефинансировать ипотеку?'
] ]
def _main(wav_file: str): def _main(wav_file: str):
# text: str = Speech().run_recognition(wav_file) # text: str = Speech().run_recognition(wav_file)
# text: str = 'Как получить деньги с заблокированной карты?' # text: str = 'Как получить деньги с заблокированной карты?'
# print(f'Text: {text}') for text in test_data:
# parse_tree: ParseTree = Syntax().get_parse_tree(text) print(f'Text: {text}')
# print(f'Parse tree:\n{parse_tree}') parse_tree: ParseTree = Syntax().get_parse_tree(text)
# terms = NLP().get_terms(parse_tree) print(f'Parse tree:\n{parse_tree}')
for terms in data: terms = NLP().get_terms(parse_tree)
print(f'Extracted terms:\n{", ".join(terms)}') print(f'Extracted terms:\n\t{", ".join(terms)}')
result: str = MyOntology().get_event_description(terms) result: str = MyOntology().get_events(terms)
print(f'Result:\n{result}') print(f'Result:\n{result}')
print() print()
if __name__ == '__main__': if __name__ == '__main__':
if len(sys.argv) < 2: if len(sys.argv) < 2:
print(f'Usage: {sys.argv[0]} FILE') print(f'Usage: {sys.argv[0]} FILE')