Skip to content

thukg/link_prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Link Prediction Models

Source code

This repository contains the following link prediction models:

  • transE
  • DistMult
  • convE
  • simplE
  • rotatE

Dataset:

  • FB13

  • FB15K

  • FB15K237

  • WN11

  • WN18

  • WN18RR

Prerequisites

  • Python3
  • Pytorch 1.0+
  • faiss(for evaluation)

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages