Skip to content

Commit

Permalink
Add description of THU-OpenSK
Browse files Browse the repository at this point in the history
  • Loading branch information
helloxcq committed Nov 2, 2022
1 parent 42a3c1d commit 0afccb5
Showing 1 changed file with 13 additions and 3 deletions.
16 changes: 13 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,5 +1,15 @@
# TensorFlow-TransX

This repository is a subproject of THU-OpenSK, and all subprojects of THU-OpenSK are as follows.

- [OpenNE](https://www.github.com/thunlp/OpenNE)
- [OpenKE](https://www.github.com/thunlp/OpenKE)
- [KB2E](https://www.github.com/thunlp/KB2E)
- [TensorFlow-Transx](https://www.github.com/thunlp/TensorFlow-Transx)
- [Fast-TransX](https://www.github.com/thunlp/Fast-TransX)
- [OpenNRE](https://www.github.com/thunlp/OpenNRE)
- [JointNRE](https://www.github.com/thunlp/JointNRE)

The implementation of TransE [1], TransH [2], TransR [3], TransD [4] for knowledge representation learning (KRL). The overall framework is based on TensorFlow. We use C++ to implement some underlying operations such as data preprocessing and negative sampling. For each specific model, it is implemented by TensorFlow with Python interfaces so that there is a convenient platform to run models on GPUs.

These codes will be gradually integrated into the new framework [[OpenKE]](https://github.com/thunlp/openke).
Expand Down Expand Up @@ -35,7 +45,7 @@ To train models based on random initialization:
1. Change class Config in transX.py

class Config(object):

def __init__(self):
...
lib.setInPath("your training data path...")
Expand All @@ -50,7 +60,7 @@ To train models based on pretrained results:
1. Change class Config in transX.py

class Config(object):

def __init__(self):
...
lib.setInPath("your training data path...")
Expand All @@ -67,7 +77,7 @@ To test your models:
1. Change class Config in transX.py

class Config(object):

def __init__(self):
...
test_lib.setInPath("your testing data path...")
Expand Down

0 comments on commit 0afccb5

Please sign in to comment.