def splitGraph(graph, random=True ,p=1/float(3)): all_edges = graph.getAllEdges() vertices = graph.vlist inc = bernoulli.rvs(p, size = len(all_edges)) train = SparseGraph(vertices) test = SparseGraph(vertices) train.addEdges(all_edges[inc==0]) test.addEdges(all_edges[inc==1]) return train, test
def splitGraph(graph, random=True, p=1 / float(3)): all_edges = graph.getAllEdges() vertices = graph.vlist inc = bernoulli.rvs(p, size=len(all_edges)) train = SparseGraph(vertices) test = SparseGraph(vertices) train.addEdges(all_edges[inc == 0]) test.addEdges(all_edges[inc == 1]) return train, test