def evaluate(node_emb, t_info): # Evaluation # Evaluate with random seed exp = Evaluation(hp.dataset) if hp.dataset == 'imdb': res = exp.evaluate_imdb(node_emb[:t_info['m']['cnt']]) elif hp.dataset == 'dblp': res = exp.evaluate_dblp(node_emb) elif hp.dataset == 'acm': res = exp.evaluate_acm(node_emb) print_dict(res)