import numpy as np
def load_graph(config):
dataset_path = config['data']['graph_pkl_filename']
graph = np.load(dataset_path)
# 将inf值填充为0
graph = np.nan_to_num(graph, nan=0.0, posinf=0.0, neginf=0.0)
return graph