9488bace208911b1bc1efb17774d8000fed1acac,examples/pytorch/ogb/deepwalk/load_dataset.py,,,#,12

Before Change

            if weighted:
                f.write(str(g.edges()[0][i].item()) + " "\
                    +str(g.edges()[1][i].item()) + " "\
                    +str(g.edata["edge_weight"][i].item()) + "\n")
            else:
                f.write(str(g.edges()[0][i].item()) + " "\
                    +str(g.edges()[1][i].item()) + " "\

After Change

        weight = list(g.edata["edge_weight"])

    print("writing...")
    start_time = time.time()
    with open(name + "-net.txt", "w") as f:
        for i in range(edge_num):
            if weighted:
                f.write(str(src[i].item()) + " "\
                    +str(tgt[i].item()) + " "\
                    +str(weight[i].item()) + "\n")
            else:
                f.write(str(src[i].item()) + " "\
                    +str(tgt[i].item()) + " "\
                    +"1\n")
    print("writing used time: %d s" % int(time.time() - start_time))

Instances