search

hpcaitech / ColossalAI-Examples

Examples of training models with hybrid parallelism using ColossalAI
Apache License 2.0
334 stars 102 forks source link

knowledge graph embedding examples - Bin Shang #39

Closed MiracleDesigner closed 2 years ago

MiracleDesigner commented 2 years ago

My name is Bin Shang. As requested by professor Yong You, I add three knowledge graph embedding examples DistMult, ComplEx and RotatE. Please check the code and feel free to contact me if you have any questions.

MiracleDesigner commented 2 years ago

您好，我已收到您的邮件。

FrankLeeeee commented 2 years ago

Hi, can you add a section Prepare Dataset in your README.md so that others know how handle the data processing. Meanwhile, have you verified the convergence of this model? If so, please attach the logs/plots here.

MiracleDesigner commented 2 years ago

Hi, can you add a section Prepare Dataset in your README.md so that others know how handle the data processing. Meanwhile, have you verified the convergence of this model? If so, please attach the logs/plots here.

OK， I would remove the dataset and add Prepare Dataset in README.md. The convergence of the model has been verified, and these three methods are the classic KGE algorithm.