cogdl.tasks.link_prediction
¶
Module Contents¶
Classes¶
Training process borrowed from KnowledgeGraphEmbedding<https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding> |
|
Functions¶
|
Save the parameters of the model and the optimizer, |
|
Write logs to checkpoint and console |
|
Print the evaluation logs |
|
|
|
|
|
|
|
|
|
|
|
-
cogdl.tasks.link_prediction.
save_model
(model, optimizer, save_variable_list, args)[source]¶ Save the parameters of the model and the optimizer, as well as some other variables such as step and learning_rate
-
class
cogdl.tasks.link_prediction.
HomoLinkPrediction
(args, dataset=None, model=None)[source]¶ Bases:
torch.nn.Module
-
class
cogdl.tasks.link_prediction.
TripleLinkPrediction
(args, dataset=None, model=None)[source]¶ Bases:
torch.nn.Module
Training process borrowed from KnowledgeGraphEmbedding<https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding>