All Projects → cai-lw → Kbgan

cai-lw / Kbgan

Licence: mit
Code for "KBGAN: Adversarial Learning for Knowledge Graph Embeddings" https://arxiv.org/abs/1711.04071

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Kbgan

Chemgan Challenge
Code for the paper: Benhenda, M. 2017. ChemGAN challenge for drug discovery: can AI reproduce natural chemical diversity? arXiv preprint arXiv:1708.08227.
Stars: ✭ 98 (-47.31%)
Mutual labels:  reinforcement-learning, generative-adversarial-network
Spiral Tensorflow
in progress
Stars: ✭ 117 (-37.1%)
Mutual labels:  reinforcement-learning, generative-adversarial-network
Simple
SimplE Embedding for Link Prediction in Knowledge Graphs
Stars: ✭ 104 (-44.09%)
Mutual labels:  knowledge-graph, knowledge-base
Deterministic Gail Pytorch
PyTorch implementation of Deterministic Generative Adversarial Imitation Learning (GAIL) for Off Policy learning
Stars: ✭ 44 (-76.34%)
Mutual labels:  reinforcement-learning, generative-adversarial-network
Tutorial Utilizing Kg
Resources for Tutorial on "Utilizing Knowledge Graphs in Text-centric Information Retrieval"
Stars: ✭ 148 (-20.43%)
Mutual labels:  knowledge-graph, knowledge-base
Stock Knowledge Graph
利用网络上公开的数据构建一个小型的证券知识图谱/知识库
Stars: ✭ 1,182 (+535.48%)
Mutual labels:  knowledge-graph, knowledge-base
Capse
A Capsule Network-based Embedding Model for Knowledge Graph Completion and Search Personalization (NAACL 2019)
Stars: ✭ 114 (-38.71%)
Mutual labels:  knowledge-graph, knowledge-base
Chatbot cn
基于金融-司法领域(兼有闲聊性质)的聊天机器人,其中的主要模块有信息抽取、NLU、NLG、知识图谱等,并且利用Django整合了前端展示,目前已经封装了nlp和kg的restful接口
Stars: ✭ 791 (+325.27%)
Mutual labels:  knowledge-graph, reinforcement-learning
Zincbase
A state of the art knowledge base
Stars: ✭ 144 (-22.58%)
Mutual labels:  knowledge-graph, knowledge-base
Hyte
EMNLP 2018: HyTE: Hyperplane-based Temporally aware Knowledge Graph Embedding
Stars: ✭ 130 (-30.11%)
Mutual labels:  knowledge-graph, knowledge-base
Gail Tf
Tensorflow implementation of generative adversarial imitation learning
Stars: ✭ 179 (-3.76%)
Mutual labels:  reinforcement-learning, generative-adversarial-network
Accel Brain Code
The purpose of this repository is to make prototypes as case study in the context of proof of concept(PoC) and research and development(R&D) that I have written in my website. The main research topics are Auto-Encoders in relation to the representation learning, the statistical machine learning for energy-based models, adversarial generation networks(GANs), Deep Reinforcement Learning such as Deep Q-Networks, semi-supervised learning, and neural network language model for natural language processing.
Stars: ✭ 166 (-10.75%)
Mutual labels:  reinforcement-learning, generative-adversarial-network
Conversational Ai
Conversational AI Reading Materials
Stars: ✭ 34 (-81.72%)
Mutual labels:  reinforcement-learning, generative-adversarial-network
Kgpolicy
Reinforced Negative Sampling over Knowledge Graph for Recommendation, WWW2020
Stars: ✭ 83 (-55.38%)
Mutual labels:  knowledge-graph, reinforcement-learning
Udacity Deep Learning Nanodegree
This is just a collection of projects that made during my DEEPLEARNING NANODEGREE by UDACITY
Stars: ✭ 15 (-91.94%)
Mutual labels:  reinforcement-learning, generative-adversarial-network
Workbase
Grakn Workbase (Knowledge IDE)
Stars: ✭ 106 (-43.01%)
Mutual labels:  knowledge-graph, knowledge-base
Pytorch Rl
PyTorch implementation of Deep Reinforcement Learning: Policy Gradient methods (TRPO, PPO, A2C) and Generative Adversarial Imitation Learning (GAIL). Fast Fisher vector product TRPO.
Stars: ✭ 658 (+253.76%)
Mutual labels:  reinforcement-learning, generative-adversarial-network
Awesome Knowledge Management
A curated list of amazingly awesome articles, people, applications, software libraries and projects related to the knowledge management space
Stars: ✭ 758 (+307.53%)
Mutual labels:  knowledge-graph, knowledge-base
Mlds2018spring
Machine Learning and having it Deep and Structured (MLDS) in 2018 spring
Stars: ✭ 124 (-33.33%)
Mutual labels:  reinforcement-learning, generative-adversarial-network
Topic Db
TopicDB is a topic maps-based semantic graph store (using PostgreSQL for persistence)
Stars: ✭ 164 (-11.83%)
Mutual labels:  knowledge-graph, knowledge-base

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

Our lab: http://nlp.cs.ucsb.edu/index.html

Dependencies

  • Python 3
  • PyTorch 0.2.0
  • PyYAML
  • nvidia-smi

PyTorch 0.2.0 is REQUIRED. Because PyTorch is not backward compatible, newer versions will NOT work. We understand that 0.2.0 is outdated, but we currently have no schedule of adding support to newer PyTorch versions.

Usage

  1. Unzip data.zip.
  2. Pretrain: python3 pretrain.py --config=config_<dataset_name>.yaml --pretrain_config=<model_name> (this will generate a pretrained model file)
  3. Adversarial train: python3 gan_train.py --config=config_<dataset_name>.yaml --g_config=<G_model_name> --d_config=<D_model_name> (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)

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].