示例#1
0
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
示例#2
0
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