Пример #1
0
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)