Skip to content

A Series of Knowledge Representation Models and Knowledge Graph Embedding Models with Pytorch.

Notifications You must be signed in to change notification settings

RuisongZhou/KGE_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

KGE_pytorch

Introduction

This is the Pytorch implementaion of some nowledge graph embedding(KGE) models. And I have test these models with popular datasets.

Implemented features

Models:

  • TransE
  • TransH
  • TransD
  • TransA
  • ConvE
  • SimplE
  • RotatE

Usage

  1. Download datasets and put it in ./data directory.
  2. Then you can run commend as follows to train/test/valid the models. All training codes can be found in ./examples/
python3 ./examples trainTransE.py

If you want to modify parameters or datasets. You can rewrite training files and overwrite raw parameters. The parameters at Config.py come from their paper.

Info

  • TransA model has a question about memory explosion

Links

Datasets:

About

A Series of Knowledge Representation Models and Knowledge Graph Embedding Models with Pytorch.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages