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)