def InitResource(version):
global database, resource, socket, listflie, tfidfmodel, tfidfdict, table_state_strategy
if version is 'v1':
listfile = 'cnn_qa_human_response_name.list'
if version is 'v2':
listfile = 'cnn_qa_human_response_name_high_app.list'
if version is 'v2.5':
listfile = 'cnn_qa_human_response_name_high_app.list'
tfidfdict = corpora.Dictionary.load(tfidfname + '.dict')
tfidfmodel = models.tfidfmodel.TfidfModel.load(tfidfname + '.tfidf')
if version is 'v3':
listfile = 'cnn_hr_v1_v2.list'
tfidfdict = corpora.Dictionary.load(tfidfname + '.dict')
tfidfmodel = models.tfidfmodel.TfidfModel.load(tfidfname + '.tfidf')
if version is 'v4':
listfile = 'cnn_hr_v1_v2_v4.list'
tfidfdict = corpora.Dictionary.load(tfidfname + '.dict')
tfidfmodel = models.tfidfmodel.TfidfModel.load(tfidfname + '.tfidf')
datalist = [line.strip() for line in open(listfile)]
database = Loader.LoadDataPair(datalist)
resource = Loader.LoadLanguageResource()
global TemplateLib, TopicLib, TreeState, Template, model
TemplateLib = Loader.LoadTemplate(template_list)
TopicLib = Loader.LoadTopic(topicfile)
TreeState, Template = Control.Init()
model = models.Doc2Vec.load('/tmp/word2vec_50')
if wizard is 2:
context = zmq.Context()
socket = context.socket(zmq.REQ)
socket.connect("tcp://localhost:5555")
with open('table_state_strategy.pkl') as f:
table_state_strategy = pickle.load(f)
def InitResource():
global database, resource
datalist = [line.strip() for line in open(listfile)]
database = Loader.LoadDataPair(datalist)
resource = Loader.LoadLanguageResource()
global TemplateLib, TopicLib, TreeState, Template
TemplateLib = Loader.LoadTemplate(template_list)
TopicLib = Loader.LoadTopic(topicfile)
TreeState, Template = Control.Init()
def InitResource(version):
global TemplateLib, TopicLib, TreeState, Template, model, init_id, joke_id, more_id, dictionary_value, turn_id, wizard, isAlltag, engaged_input, engagement_mode
global q_table, database, resource, socket, listflie, tfidfmodel, tfidfdict, table_state_strategy
database = {}
resource = {}
listfile = None
init_id = 0
joke_id = 0
more_id = 0
wizard = 3
isAlltag = 0
turn_id = 0
engaged_input = []
engagement_mode = 0
rescource_root = 'resource'
template_list = [
'template/template_new.txt', 'template/template_end.txt',
'template/template_open.txt', 'template/template_expand.txt',
'template/template_init.txt', 'template/template_joke.txt',
'template/template_back.txt', 'template/template_more.txt'
]
template_list = [path.join(rescource_root, name) for name in template_list]
topicfile = path.join(rescource_root, 'topic.txt')
tfidfname = 'tfidf_reference'
with open('dictionary_value.pkl') as f:
dictionary_value = pickle.load(f)
if version is 'v1':
listfile = 'cnn_qa_human_response_name.list'
elif version is 'v2':
listfile = 'cnn_qa_human_response_name_high_app.list'
elif version is 'v2.5':
listfile = 'cnn_qa_human_response_name_high_app.list'
tfidfdict = corpora.Dictionary.load(tfidfname + '.dict')
tfidfmodel = models.tfidfmodel.TfidfModel.load(tfidfname + '.tfidf')
elif version is 'v3':
listfile = 'cnn_hr_v1_v2.list'
tfidfdict = corpora.Dictionary.load(tfidfname + '.dict')
tfidfmodel = models.tfidfmodel.TfidfModel.load(tfidfname + '.tfidf')
elif version is 'v4':
listfile = 'cnn_hr_v1_v2_v4.list'
tfidfdict = corpora.Dictionary.load(tfidfname + '.dict')
tfidfmodel = models.tfidfmodel.TfidfModel.load(tfidfname + '.tfidf')
datalist = [line.strip() for line in open(listfile)]
q_table = pickle.load(open('q_table.pkl'))
database = Loader.LoadDataPair(datalist)
resource = Loader.LoadLanguageResource()
TemplateLib = Loader.LoadTemplate(template_list)
TopicLib = Loader.LoadTopic(topicfile)
TreeState, Template = Control.Init()
model = models.Doc2Vec.load('/tmp/word2vec_50')
if wizard is 2:
context = zmq.Context()
socket = context.socket(zmq.REQ)
socket.connect("tcp://localhost:5555")
with open('table_state_strategy.pkl') as f:
table_state_strategy = pickle.load(f)
def InitResource():
global database, resource, socket
datalist=[line.strip() for line in open(listfile)]
database = Loader.LoadDataPair(datalist)
resource = Loader.LoadLanguageResource()
global TemplateLib, TopicLib, TreeState, Template
TemplateLib = Loader.LoadTemplate(template_list)
TopicLib = Loader.LoadTopic(topicfile)
TreeState, Template = Control.Init()
context= zmq.Context()
#print("connectting to server")
socket = context.socket(zmq.REQ)
socket.connect("tcp://localhost:5555")
