diff --git a/EP/Cora_node_classification/cora.py b/EP/Cora_node_classification/cora.py index 1eed9f8dc3a0835a6775d9f9052754f4d5a02327..c1a3d95f386e361be0cc7d7c51dc5a854769ca39 100644 --- a/EP/Cora_node_classification/cora.py +++ b/EP/Cora_node_classification/cora.py @@ -117,7 +117,7 @@ def write_pickle_graph_file(path_nodes="/cora_data/cora.content", path_edges="/c with open(path_output_graph + "graph.pkl", "wb") as output: pkl.dump(g, output) -def read_pickle_graph(path = "graph.pkl"): +def read_pickle_graph(path = "graph.pkl"): #will be used on node_classification.py for accessing the graph with open(path, 'rb') as f: graph = pkl.load(f) return graph