awslabs / dgl-ke

High performance, easy-to-use, and scalable package for learning large-scale knowledge graph embeddings.
https://dglke.dgl.ai/doc/
Apache License 2.0
1.28k stars 196 forks source link

[Feature] Edge Importance score support #149

Closed classicsong closed 4 years ago

classicsong commented 4 years ago

Support training a knowledge graph with weighted edges (each edge has an importance score).

The example can be found in examples/wn18_weighted/

148