def __init__(self):
gl.set_NUM_OF_RELATIONS(self.NUM_OF_RELATIONS)
gl.set_LSTM_DIM(self.LSTM_DIM)
gl.set_EMBEDDING_DIM(self.EMBEDDING_DIM)
gl.set_ques_maxlen(self.ques_maxlen)
gl.set_relation_maxlen(self.rela_maxlen)
gl.set_NUM_FILTERS(self.NUM_FILTERS)
self.CLASS_INDEX = self.load_CLASS_INDEX()
self.relation_vec, self.wd_idx, self.embedding_matrix = self.prepareWork(
)
self.model = self.loadCNNModelFromFile()
inpute_question = "创始人 是 谁"
self.predicated_quick(inpute_question)
import globalvar as gl
from preprocess.data_preprocessor import preprocess, generateWord2VectorMatrix, loadEmbeddingsIndex
#全局变量
gl.set_train_rela_files("train_case_rela.txt")
gl.set_train_ques_file("train_case_ques.txt")
gl.set_train_label_file("train_case_label.txt")
gl.set_test_rela_files("test_case_rela.txt")
gl.set_test_ques_file("test_case_ques.txt")
gl.set_test_label_file("test_case_label.txt")
gl.set_preprocessWordVector_files("TencentPreTrain.txt")
gl.set_preprocessWordVector_path("/data/zjy/")
gl.set_MAX_NB_WORDS(30)
gl.set_EMBEDDING_DIM(200)
gl.set_LSTM_DIM(150)
train_rela_files = gl.get_train_rela_files()
train_ques_file = gl.get_train_ques_file()
train_label_file = gl.get_train_label_file()
test_rela_files = gl.get_test_rela_files()
test_ques_file = gl.get_test_ques_file()
test_label_file = gl.get_test_label_file()
preprocessWordVector_files = gl.get_preprocessWordVector_files()
preprocessWordVector_path = gl.get_preprocessWordVector_path()
MAX_NB_WORDS = gl.get_MAX_NB_WORDS()
EMBEDDING_DIM = gl.get_EMBEDDING_DIM()
LSTM_DIM = gl.get_LSTM_DIM()
gl.set_NUM_FILTERS(150)