search

dice-group / dice-embeddings

Hardware-agnostic Framework for Large-scale Knowledge Graph Embeddings
MIT License
44 stars 12 forks source link

Expanding to other datasets #240

Open akalino opened 6 months ago

akalino commented 6 months ago

@Demirrr Any interest in creating pretrained model checkpoints for other large-scale graphs? In particular, aside from YAGO/WN18RR/FB15K-237, there are several other graphs that are used for alignment or relation extraction benchmarks (see ]https://hadyelsahar.github.io/t-rex/). In your opinion, would pretrained models for these other graphs be a useful addition to this repo or documentation?

Demirrr commented 6 months ago

Dear @akalino

Thank you for your question.

Frankly, I do not know much about the T-Rex dataset. Given that the dataset comes in the ttle format, I reckon we can easily provide few pretrained models. Let me take a deeper look at the dataset and train few models :)