Add test data

This commit is contained in:
Aleksey Filippov 2023-06-02 17:48:33 +04:00
parent 50e36be6d0
commit f1c4f5ad41

60
main.py
View File

@ -3,21 +3,59 @@
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 data = [
from src.syntax import Syntax ['получить кредит обучение'],
['получить кредит образование'],
['купить товар рассрочка'],
['кредит ставка ипотека'],
['самая низкий ставка кредиту'],
['оформить кредит'],
['ставка кредит'],
['ипотека кредит ставка', 'какой ставка'],
['получить деньга карта', 'заблокировать'],
['снять деньга карта'],
['какой баланс карта'],
['лимит карта'],
['проходить оплата', 'быть деньга'],
['деньга карта'],
['деньга счёт'],
['разблокировать карта'],
['заблокировать карта'],
['какой условие ипотека'],
['какой условие кредит жильё'],
['какой срок кредит жильё'],
['какой срок ипотека'],
['потерять карта'],
['застрять банкомат'],
['украсть карта'],
['какой остаток карта'],
['купить дом'],
['купить дом'],
['купить товар кредит'],
['срок ипотека'],
['купить дом кредит'],
['снизить ставка кредиту'],
['рефинансировать кредит'],
['снизить ставка кредиту'],
['ли друг банка банк, перевести банк'],
['провести рефинансирование кредит'],
['рефинансировать ипотека']
]
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}') # print(f'Text: {text}')
parse_tree: ParseTree = Syntax().get_parse_tree(text) # parse_tree: ParseTree = Syntax().get_parse_tree(text)
print(f'Parse tree:\n{parse_tree}') # print(f'Parse tree:\n{parse_tree}')
terms = NLP().get_terms(parse_tree) # terms = NLP().get_terms(parse_tree)
print(f'Extracted terms:\n{", ".join(terms)}') for terms in data:
result: str = MyOntology().get_event_description(terms) print(f'Extracted terms:\n{", ".join(terms)}')
print(f'Result:\n{result}') result: str = MyOntology().get_event_description(terms)
print(f'Result:\n{result}')
print()
if __name__ == '__main__': if __name__ == '__main__':