divide_data (input_list, division_rate)
|
|
randomly_choose_false_edges (nodes, true_edges, num)
|
|
gen_node_pairs (train_data, test_data, negative_ratio=5)
|
|
get_score (embs, node1, node2)
|
|
evaluate (embs, true_edges, false_edges)
|
|
select_task (model_name=None, model=None)
|
|