KBGAN-PyTorch-v0.4.1

KO | EN

gitlite — search

#python #ai #llm #go #machine-learning #deep-learning #docker #nodejs #windows #cli #kubernetes #pytorch #ios #openai

KBGAN-PyTorch-v0.4.1

(★ 12)

No description available.

KBGAN-PyTorch-v0.4.1 Latest Version Download

Download Latest Version (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# KBGAN
## This repo is the pytorch-v0.4.1 implement version for KBGAN.

> The origin repo: https://github.com/cai-lw/KBGAN  
> Liwei Cai and William Yang Wang, "KBGAN: Adversarial Learning for Knowledge Graph Embeddings", in *Proceedings of The 16th Annual Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies (NAACL HLT 2018)*.    
> Paper: https://arxiv.org/abs/1711.04071

## Dependencies
* Python 3.6
* PyTorch 0.4.1
* PyYAML
* nvidia-smi

## Usage
- - -
* Pretrain(for example):   
python pretrain.py --config=config_fb15k237.yaml --pretrain_config=TransE  
python pretrain.py --config=config_fb15k237.yaml --pretrain_config=DistMult  
(this will generate a pretrained model file)
* Adversarial train:  
 python gan_train.py --config=config_fb15k237.yaml --g_config=TransE --d_config=DistMult  
(make sure that G model and D model are both pretrained)   
- - -
Feel free to explore and modify parameters in config files. Default parameters are those used in experiments reported in the paper.  
Decrease **test_batch_size** in config files if you experience GPU memory exhaustion. (this would make the program runs slower, but would not affect the test result)

KBGAN-PyTorch-v0.4.1

KBGAN-PyTorch-v0.4.1 Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

KBGAN-PyTorch-v0.4.1

KBGAN-PyTorch-v0.4.1 Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Download README