def SetConf(self, ConfIn):
ObjObjEdgeFeatureExtractorC.SetConf(self, ConfIn)
Word2VecIn = self.conf.GetConf('word2vecin')
logging.info('start load word2vec from [%s]', Word2VecIn)
self.Word2VecModel = gensim.models.Word2Vec.load_word2vec_format(
Word2VecIn)
logging.info('loaded')
def Init(self):
ObjObjEdgeFeatureExtractorC.Init(self)
self.FeatureName += 'Type'
self.lFeatureName = [
self.FeatureName + item
for item in ['HasSame', 'SameTypeFrac', 'SameNotable']
]
def Init(self):
ObjObjEdgeFeatureExtractorC.Init(self)
self.FeatureName += 'KG'
self.lFeatureDim = [
self.FeatureName + item
for item in ['Connected', 'HasCommonNeighbor']
]
def Init(self):
ObjObjEdgeFeatureExtractorC.Init(self)
self.FeatureName += 'PreCalc'
self.PreCalcFileInName = ""
self.lSimName = [] #read from PreCalcFileInName's referenced files
self.lPreCalcDir = [] #the dir contains query's obj sim dict
self.lDirected = [] #whether is directed
self.lhQueryObjPairSim = [] #l [{}for each pre calc sim ] {query -> hobjpairsim}}
def Init(self):
ObjObjEdgeFeatureExtractorC.Init(self)
self.FeatureName += 'TextSim'
# self.lObjField = ['name','desp','alias']
# self.lFieldSimMetric = ['coor','js','cosine']
self.lObjField = ['desp']
self.lFieldSimMetric = ['cosine']
self.CtfCenter = TermCtfC()
self.TermCtfIn = ""
def ShowConf():
ObjObjEdgeFeatureExtractorC.ShowConf()
print 'precalcsimfile'
def SetConf(self, ConfIn):
ObjObjEdgeFeatureExtractorC.SetConf(self, ConfIn)
self.PreCalcFileInName = self.conf.GetConf('precalcsimfile')
self.SetPreCalcSim()
def ShowConf():
ObjObjEdgeFeatureExtractorC.ShowConf()
print 'termctf'
def SetConf(self, ConfIn):
ObjObjEdgeFeatureExtractorC.SetConf(self, ConfIn)
self.TermCtfIn = self.conf.GetConf('termctf')
self.CtfCenter = TermCtfC(self.TermCtfIn)
def ShowConf():
ObjObjEdgeFeatureExtractorC.ShowConf()
print 'word2vecin'
def Init(self):
ObjObjEdgeFeatureExtractorC.Init(self)
self.FeatureName += 'EmbSim'
self.Word2VecModel = None
self.lFeatureName = [self.FeatureName + 'Cosine']