Update test data
This commit is contained in:
parent
3eb01b064c
commit
ef33282cbb
92
main.py
92
main.py
@ -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')
|
||||||
|
Loading…
Reference in New Issue
Block a user