_logger = log_helper.get_logger()

def __init__(self, name, node_id):

self.node_id = str(node_id)

self.dir = utils.get_data_dir(name)

if utils.graph_directness[name]:

self.graph = nx.DiGraph()

else:

self.graph = nx.Graph()

def get_fname(self, ext):

fname = self.node_id + '.' + ext

return os.path.join(self.dir, fname)

def circles_reader(self):

fname = self.get_fname('circles')

assert (os.path.isfile(fname))

with open(fname) as f:

for line in f:

id_list = line.split()[1:]

yield [int(item) for item in id_list]

def graph_generator(self):

pickle_name = self.get_fname('pickle')

if os.path.isfile(pickle_name):

self._logger.info('load {}'.format(pickle_name))

with open(pickle_name, 'rb') as handler:

self.graph = pickle.load(handler)

else:

self._logger.info('create {}'.format(pickle_name))

edges = self.edges_reader()

self.graph.add_edges_from(edges)

for n in self.graph.nodes():

self.graph.add_edge(self.node_id, n)

with open(pickle_name, 'wb') as handler:

pickle.dump(self.graph, handler)

def save_figure(self, fmt='pdf'):

pos = nx.spectral_layout(self.graph)

nx.draw_networkx(self.graph, pos)

fname = self.get_fname(fmt)

plt.savefig(fname)

def edges_reader(self):

fname = self.get_fname('edges')

assert (os.path.isfile(fname))

edge_list = []

with open(fname) as f:

for line in f:

node_list = [int(elem) for elem in line.split()]

edge_list.append(tuple(node_list))

return edge_list

def feat_reader(self):

fname = self.get_fname('feat')

assert (os.path.isfile(fname))

with open(fname) as f:

for line in f:

digit_str_list = line.split()

digits = [int(elem) for elem in digit_str_list][1:]