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)