Add test data
This commit is contained in:
parent
50e36be6d0
commit
f1c4f5ad41
60
main.py
60
main.py
@ -3,21 +3,59 @@
|
||||
import sys
|
||||
|
||||
from src.myontology import MyOntology
|
||||
from src.nlp import NLP
|
||||
from src.parse_tree.parse_tree import ParseTree
|
||||
from src.syntax import Syntax
|
||||
|
||||
data = [
|
||||
['получить кредит обучение'],
|
||||
['получить кредит образование'],
|
||||
['купить товар рассрочка'],
|
||||
['кредит ставка ипотека'],
|
||||
['самая низкий ставка кредиту'],
|
||||
['оформить кредит'],
|
||||
['ставка кредит'],
|
||||
['ипотека кредит ставка', 'какой ставка'],
|
||||
['получить деньга карта', 'заблокировать'],
|
||||
['снять деньга карта'],
|
||||
['какой баланс карта'],
|
||||
['лимит карта'],
|
||||
['проходить оплата', 'быть деньга'],
|
||||
['деньга карта'],
|
||||
['деньга счёт'],
|
||||
['разблокировать карта'],
|
||||
['заблокировать карта'],
|
||||
['какой условие ипотека'],
|
||||
['какой условие кредит жильё'],
|
||||
['какой срок кредит жильё'],
|
||||
['какой срок ипотека'],
|
||||
['потерять карта'],
|
||||
['застрять банкомат'],
|
||||
['украсть карта'],
|
||||
['какой остаток карта'],
|
||||
['купить дом'],
|
||||
['купить дом'],
|
||||
['купить товар кредит'],
|
||||
['срок ипотека'],
|
||||
['купить дом кредит'],
|
||||
['снизить ставка кредиту'],
|
||||
['рефинансировать кредит'],
|
||||
['снизить ставка кредиту'],
|
||||
['ли друг банка банк, перевести банк'],
|
||||
['провести рефинансирование кредит'],
|
||||
['рефинансировать ипотека']
|
||||
]
|
||||
|
||||
|
||||
def _main(wav_file: str):
|
||||
# text: str = Speech().run_recognition(wav_file)
|
||||
text: str = 'Как получить деньги с заблокированной карты?'
|
||||
print(f'Text: {text}')
|
||||
parse_tree: ParseTree = Syntax().get_parse_tree(text)
|
||||
print(f'Parse tree:\n{parse_tree}')
|
||||
terms = NLP().get_terms(parse_tree)
|
||||
print(f'Extracted terms:\n{", ".join(terms)}')
|
||||
result: str = MyOntology().get_event_description(terms)
|
||||
print(f'Result:\n{result}')
|
||||
# text: str = 'Как получить деньги с заблокированной карты?'
|
||||
# print(f'Text: {text}')
|
||||
# parse_tree: ParseTree = Syntax().get_parse_tree(text)
|
||||
# print(f'Parse tree:\n{parse_tree}')
|
||||
# terms = NLP().get_terms(parse_tree)
|
||||
for terms in data:
|
||||
print(f'Extracted terms:\n{", ".join(terms)}')
|
||||
result: str = MyOntology().get_event_description(terms)
|
||||
print(f'Result:\n{result}')
|
||||
print()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
Loading…
Reference in New Issue
Block a user