def __init__(self, img_size=32, stacks=4, skips=4, return_seq=False):
env = gym_super_mario_bros.make('SuperMarioBros-v2')
self.env = JoypadSpace(env, SIMPLE_MOVEMENT)
self.preprocess = Preprocess(img_size, stacks, return_seq)
self.skips = skips
self.action_space = self.env.action_space
self.observation_space = (img_size, img_size, stacks)
def __init__(self, img_size=32, stacks=4, skips=4, return_seq=False):
self.env = gym.make('Pong-v0')
self.preprocess = Preprocess(img_size, stacks, return_seq)
self.skips = skips
self.pong_action = {0: 0, 1: 2, 2: 3}
self.action_space = self.env.action_space
self.action_space.n = 3
self.observation_space = (img_size, img_size, stacks)
from models.intent.IntentModel import IntentModel
from utils.Preprocess import Preprocess
p = Preprocess(word2index_dic='train_tools/dict/chatbot_dict.bin',
userdic='../Tokenizing/user_dic.txt')
intent = IntentModel(model_name='models/intent/intent_model.h5', proprocess=p)
query = "안녕하세요."
intent_predict = intent.predict_class(query)
print(intent_predict)
import sys
sys.path.append('/content/drive/MyDrive/deep-chatbot/')
import pickle
from utils.Preprocess import Preprocess
# 단어 사전 불러오기
f = open("drive/MyDrive/deep-chatbot/train_tools/dict/chatbot_dict.bin", "rb")
word_index = pickle.load(f)
f.close()
sent = "데이터베이스가 뭔지 알려줘"
# 전처리 객체 생성
p = Preprocess(userdic='drive/MyDrive/deep-chatbot/utils/user_dic.tsv')
# 형태소분석기 실행
pos = p.pos(sent)
# 품사 태그 없이 키워드 출력
keywords = p.get_keywords(pos, without_tag=True)
for word in keywords:
try:
print(word, word_index[word])
except KeyError:
# 해당 단어가 사전에 없는 경우, OOV 처리
print(word, word_index['OOV'])
