predict = intent.predict_class(query)
intent_name = intent.labels[predict]
# 개체명 인식
from models.ner.NerModel import NerModel
ner = NerModel(model_name='models/ner/ner_model.h5', proprocess=p)
predicts = ner.predict(query)
ner_tags = ner.predict_tags(query)
print("질문 : ", query)
print("=" * 100)
print("의도 파악 : ", intent_name)
print("개체명 인식 : ", predicts)
print("답변 검색에 필요한 NER 태그 : ", ner_tags)
print("=" * 100)
# 답변 검색
from utils.FindAnswer import FindAnswer
try:
f = FindAnswer(db)
answer_text, answer_image = f.search(intent_name, ner_tags)
answer = f.tag_to_word(predicts, answer_text)
except:
answer = "죄송해요 무슨 말인지 모르겠어요"
print("답변 : ", answer)
db.close() # 디비 연결 끊음
preprocess=p)
predict = intent.predict_class(query)
intent_name = intent.labels[predict]
# 개체명 인식
from models.ner.NerModel import NerModel
ner = NerModel(model_name='../models/ner/ner_model.h5', preprocess=p)
predicts = ner.predict(query)
ner_tags = ner.predict_tags(query)
print("질문 : ", query)
print("=" * 40)
print("의도 파악 : ", intent_name)
print("개체명 인식 : ", predicts)
print("답변 검색에 필요한 NER 태그", ner_tags)
print("=" * 40)
# 답변 검색
from utils.FindAnswer import FindAnswer
try:
f = FindAnswer(db)
answer_text, answer_image = f.search(intent_name, ner_tags)
answer = f.tag_to_word(predicts, answer_text)
except:
answer = "죄송해요, 무슨 말인지 모르겠어요."
print("답변 : ", answer)
db.close() # DB 연결 끊음