snap-stanford/GraphRNN

Trouble running evaluate.py

gshartnett opened this issue · 2 comments

I am having trouble running evaluate.py after training the graphRNN. The issue seems to be the function load_graph_list in utils.py (starting at line 459):

# load a list of graphs
def load_graph_list(fname,is_real=True):

    for i in range(len(graph_list)):
        edges_with_selfloops = graph_list[i].selfloop_edges()
        if len(edges_with_selfloops)>0:
            graph_list[i].remove_edges_from(edges_with_selfloops)
        if is_real:
            graph_list[i] = max(nx.connected_component_subgraphs(graph_list[i]), key=len)
            graph_list[i] = nx.convert_node_labels_to_integers(graph_list[i])
        else:
            graph_list[i] = pick_connected_component_new(graph_list[i])
    return graph_list

This function calls the undefined list graph_list, and it does not load any graphs from file. I suspect that the code listed here was intended for a different function.

I have the same problem.

Hi, have you solved the problem?