Contents:
tasks.link_prediction
LinkPrediction
divide_data(input_list, division_rate)
divide_data
randomly_choose_false_edges(nodes, true_edges, num)
randomly_choose_false_edges
gen_node_pairs(train_data, test_data, negative_ratio=5)
gen_node_pairs
get_score(embs, node1, node2)
get_score
evaluate(embs, true_edges, false_edges)
evaluate
tasks.link_prediction.
Bases: tasks.BaseTask
tasks.BaseTask
add_args
Add task-specific arguments to the parser.
train