|
|
@@ -58,8 +58,7 @@ class CodeNamePredict():
|
|
|
self.BiRNN_UNITS = 200
|
|
|
else:
|
|
|
self.BiRNN_UNITS = BiRNN_UNITS
|
|
|
- # self.filepath = os.path.dirname(__file__)+"/../projectCode/models/model_project_"+str(self.EMBED_DIM)+"_"+str(self.BiRNN_UNITS)+".hdf5"
|
|
|
- self.filepath = os.path.dirname(__file__)+"codename_savedmodel_tf/saved_model.pb"
|
|
|
+ self.filepath = os.path.dirname(__file__)+"/../projectCode/models/model_project_"+str(self.EMBED_DIM)+"_"+str(self.BiRNN_UNITS)+".hdf5"
|
|
|
#self.filepath = "../projectCode/models/model_project_60_200_200ep017-loss6.456-val_loss7.852-val_acc0.969.hdf5"
|
|
|
self.filepath_code = os.path.dirname(__file__)+"/../projectCode/models/model_code.hdf5"
|
|
|
vocabpath = os.path.dirname(__file__)+"/codename_vocab.pk"
|
|
|
@@ -74,10 +73,8 @@ class CodeNamePredict():
|
|
|
id_PN_B = self.class_labels.index("PN_B")
|
|
|
id_PN_M = self.class_labels.index("PN_M")
|
|
|
id_PN_E = self.class_labels.index("PN_E")
|
|
|
- # self.PC_pattern = re.compile(str(id_PC_B)+str(id_PC_M)+"+"+str(id_PC_E)+"?")
|
|
|
- self.PC_pattern = re.compile(str(4)+str(5)+"*"+str(6))
|
|
|
- # self.PN_pattern = re.compile(str(id_PN_B)+str(id_PN_M)+"+"+str(id_PN_E)+"?")
|
|
|
- self.PN_pattern = re.compile(str(1)+str(2)+"*"+str(3))
|
|
|
+ self.PC_pattern = re.compile(str(id_PC_B)+str(id_PC_M)+"*"+str(id_PC_E))
|
|
|
+ self.PN_pattern = re.compile(str(id_PN_B)+str(id_PN_M)+"*"+str(id_PN_E))
|
|
|
print("pc",self.PC_pattern)
|
|
|
print("pn",self.PN_pattern)
|
|
|
self.word2index = dict((w,i) for i,w in enumerate(np.array(self.vocab)))
|
