def setUpClass(cls):
cls.graph = loader.load_raw_graph('tests/data/12831.edges')
parser.add_argument('-i', '--input', required=True)
parser.add_argument('-o', '--output', default=None)
parser.add_argument('-f',
'--format',
default='gml',
help='|'.join(list(FORMATS.keys())))
parser.add_argument('-co',
'--converter',
default=None,
help='|'.join(list(CONVERTERS.keys())))
return parser.parse_args()
if __name__ == "__main__":
args = parse_arguments()
logzero.logfile('logs/format_graph.log', mode='w')
if args.format not in FORMATS.keys():
raise ValueError('Invalid format `%s`' % args.format)
g = load_raw_graph(args.input)
if args.converter is not None:
g = CONVERTERS[args.converter](g)
if args.output is None:
args.output = '.'.join(
args.input.split('.')[:-1]) + '.%s' % args.format
logger.info('Saving conversion to %s' % args.output)
FORMATS[args.format](g, args.output, stringizer=str)
parser.add_argument('-i', '--logdir', required=True)
parser.add_argument('-o', '--outdir', default=None)
parser.add_argument('-f', '--feats', default=None)
parser.add_argument('-fn', '--featnames', default=None)
return parser.parse_args()
if __name__ == "__main__":
args = parse_arguments()
logzero.logfile('logs/analyze_results.log', mode='w')
if args.outdir is None:
args.outdir = args.logdir
os.makedirs(args.outdir, exist_ok=True)
g = loader.load_raw_graph(os.path.join(args.logdir, 'graph.txt'))
has_feats = args.feats is not None and args.featnames is not None
feat_names = []
if has_feats:
logger.info('Loading features')
feat_names = loader.add_node_feats(g, args.feats, args.featnames)
clpath = os.path.join(args.logdir, 'clusters.txt')
clusters = loader.load_clusters(clpath)
metrics, props = OrderedDict(), []
FEAT_K = 5
# Node count
logger.info('Logging node count')
def setUpClass(cls):
cls.graph = loader.load_raw_graph('tests/data/477094958.edges')
cls.graph = conversions.naive_conversion(cls.graph)
def test_graph_load(self):
g = loader.load_raw_graph('tests/data/12831.edges')
self.assertEqual(2478, len(g.edges))
self.assertTrue(g.has_edge(398874773, 652193))
def test_graph_feats(self):
g = loader.load_raw_graph('tests/data/12831.edges')
feats = loader.add_node_feats(g, 'tests/data/12831.feat',
'tests/data/12831.featnames')
self.assertEqual(len(feats), 1364)