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")