def createGraph(numNodes = 10): #numNodes = int(NumNodes) G_fn = "bench_concomp"+str(numNodes) nodeProb = 0.4 randGr = nx.binomial_graph(numNodes,nodeProb,directed=False) adjMat = [ [ 0 for i in range(numNodes) ] for j in range(numNodes) ] for row in range (len(randGr.adj.items())): for ind in ((randGr.adj.items()[row])[1]).keys(): adjMat[row][ind] = 1 # Make equivalent csc_matrix Gsparse = csc_matrix(adjMat, dtype=float64) if not os.path.exists(os.path.join("bench",str(numNodes))): os.makedirs(os.path.join("bench",str(numNodes))) sio.savemat(os.path.join("bench",str(numNodes) ,G_fn),{'fibergraph': Gsparse}, appendmat = True) inv = evaluate_graph(randGr, -1) printUTIL(inv) writeUTIL(inv)
def createGraph(numNodes, nodeProb): """ Create a graph and compute and save """ print "Creating random graph with node connectivity probability = %f ...." % nodeProb G_fn = "bench_concomp"+str(numNodes) randGr = nx.binomial_graph(numNodes, nodeProb, directed=False) adjMat = [ [ 0 for i in range(numNodes) ] for j in range(numNodes) ] for row in range (len(randGr.adj.items())): for ind in ((randGr.adj.items()[row])[1]).keys(): adjMat[row][ind] = 1 # Make equivalent csc_matrix Gsparse = csc_matrix(adjMat, dtype=float64) gdir = os.path.join("bench",str(numNodes)) if not os.path.exists(gdir): os.makedirs(gdir) sio.savemat(os.path.join(gdir ,G_fn),{"fibergraph": Gsparse}, appendmat = True) print "Your graph is saved in %s ...\n" % os.path.abspath(gdir) inv = evaluate_graph(randGr, -1) printUTIL(inv) writeUTIL(inv)
def createER(numNodes, nodeProb): """ Create an ER graph, compute invariants from goldstd and write to disk as scipy csc """ print "Creating ER graph with node connectivity probability = %f ...." % nodeProb G_fn = "bench_concomp"+str(numNodes) g = nx.erdos_renyi_graph(numNodes, nodeProb, directed=False) gs = nx.to_scipy_sparse_matrix(g, format="csc", dtype="float32") gdir = os.path.join("bench",str(numNodes)) if not os.path.exists(gdir): os.makedirs(gdir) sio.savemat(os.path.join(gdir ,G_fn),{"fibergraph": Gsparse}, appendmat = True) print "Your graph is saved in %s ...\n" % os.path.abspath(gdir) inv = evaluate_graph(g, -1) printUTIL(inv) writeUTIL(inv)