def learn_embedding(self,
graph=None,
edge_f=None,
is_weighted=False,
no_python=False):
args = ["gem/c_exe/node2vec"]
if not graph and not edge_f:
raise Exception('graph/edge_f needed')
if edge_f:
graph = graph_util.loadGraphFromEdgeListTxt(edge_f)
graph_util.saveGraphToEdgeListTxtn2v(graph, 'embs/n2v_tempGraph.graph')
args.append("-i:embs/n2v_tempGraph.graph")
args.append("-o:embs/n2v_tempGraph.emb")
args.append("-d:%d" % self._d)
args.append("-l:%d" % self._walkLength)
args.append("-r:%d" % self._numWalks)
args.append("-k:%d" % self._contextSize)
args.append("-e:%d" % self._max_iter)
args.append("-p:%f" % self._return_p)
args.append("-q:%f" % self._inout_p)
args.append("-v")
args.append("-dr")
args.append("-w")
t1 = time()
try:
call(args)
except Exception as e:
print(str(e))
raise Exception(
'./node2vec not found. Please compile snap, place node2vec in the path and grant executable permission'
)
t2 = time()
self._X = graph_util.loadEmbedding('embs/n2v_tempGraph.emb')
return self._X, (t2 - t1)
def learn_embedding(self,
graph=None,
edge_f=None,
is_weighted=False,
no_python=False):
args = ["node2vec"]
if not graph and not edge_f:
raise Exception('graph/edge_f needed')
if graph is None:
graph = graph_util.loadGraphFromEdgeListTxt(edge_f, directed=True)
print("node2vec graph")
print(graph.edges()[:3])
# print(graph)
graph_util.saveGraphToEdgeListTxtn2v(graph, 'tempGraph.graph')
args.append("-i:tempGraph.graph")
args.append("-o:tempGraph.emb")
args.append("-d:%d" % self._d)
args.append("-l:%d" % self._walk_len)
args.append("-r:%d" % self._num_walks)
args.append("-k:%d" % self._con_size)
args.append("-e:%d" % self._max_iter)
args.append("-p:%f" % self._ret_p)
args.append("-q:%f" % self._inout_p)
args.append("-v")
args.append("-dr")
args.append("-w")
t1 = time()
try:
call(args)
except Exception as e:
print(str(e))
raise Exception(
'./node2vec not found. Please compile snap, place node2vec in the system path and grant executable permission'
)
self._X = graph_util.loadEmbedding('tempGraph.emb')
t2 = time()
print('len graph edges')
print(len(graph.nodes()))
print('embedding vectors number')
print(len(self._X))
print('GUESS embedding node2vc')
for i in range(len(self._X)):
print(str(graph.nodes()[i]) + " " + str(self._X[i]))
# print (self._X)
return self._X, (t2 - t1)
def learn_embedding(self,
graph=None,
edge_f=None,
is_weighted=False,
no_python=False):
args = ["gem/c_exe/node2vec"]
if not graph and not edge_f:
raise Exception('graph/edge_f needed')
if edge_f:
graph = graph_util.loadGraphFromEdgeListTxt(edge_f)
graphFileName = 'gem/intermediate/%s_n2v.graph' % self._data_set
embFileName = 'gem/intermediate/%s_%d_n2v.emb' % (self._data_set,
self._d)
try:
f = open(graphFileName, 'r')
f.close()
except IOError:
graph_util.saveGraphToEdgeListTxtn2v(graph, graphFileName)
args.append("-i:%s" % graphFileName)
args.append("-o:%s" % embFileName)
args.append("-d:%d" % self._d)
args.append("-l:%d" % self._walk_len)
args.append("-r:%d" % self._num_walks)
args.append("-k:%d" % self._con_size)
args.append("-e:%d" % self._max_iter)
args.append("-p:%f" % self._ret_p)
args.append("-q:%f" % self._inout_p)
args.append("-v")
args.append("-dr")
args.append("-w")
t1 = time()
try:
call(args)
except Exception as e:
print(str(e))
raise Exception(
'./node2vec not found. Please compile snap, place node2vec in the path and grant executable permission'
)
self._X = graph_util.loadEmbedding(embFileName)
t2 = time()
call(["rm", embFileName])
return self._X, (t2 - t1)
