ilya/VKR_2026
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix errors

Browse Source
This commit is contained in:
podiukov.iv
2026-04-24 17:57:35 +05:00
parent 64460d91b4
commit d38b9337c9
3 changed files with 25 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
main.py
View File

@@ -2,7 +2,7 @@
from modules.NER import NER
from modules.paraGenerator import ParaphraseGenerator
#from modules.validator import validator
from modules.validator import validator
ner = NER()
pg = ParaphraseGenerator()
@@ -10,22 +10,11 @@ pg = ParaphraseGenerator()
srcText = 'Добрый день, я, Сидоров Иван Иванович. Прошу перевести сто тысяч рублей Якову Петру Игнатьевичу в Москву.'
def main(srcText):
# поиск сущностей
srcEntities = ner.extract_entities(srcText)
print(srcEntities)
# генерация парафраза
paraphrase = pg.generate(srcText, srcEntities)
print(paraphrase)
# поиск сущностей в парафразе
paraEntities = ner.extract_entities(paraphrase)
print(paraEntities)
# Валидация
# return validator(srcText, paraphrase, srcEntities, paraEntities)
srcEntities = ner.extract_entities(srcText) # поиск сущностей
paraphrase = pg.generate(srcText, srcEntities) # генерация парафраза
paraEntities = ner.extract_entities(paraphrase) # поиск сущностей в парафразе
return validator(srcText, paraphrase, srcEntities, paraEntities) # валидация
result = main(srcText)
print(result)
print(f'ИСХОДНЫЙ ТЕКСТ: {srcText}')
print(f'СГЕНЕРИРОВАННЫЙ ТЕКСТ: {result}')