https://github.com/BookmanHan/Embedding
Tip revision: c818fc3a398cc2edb0e0a1d7536aa7a56a12a07d authored by XiaoHan on 19 April 2018, 01:41:10 UTC
Update README.md
Update README.md
Tip revision: c818fc3
README.md
# Embedding
This Project is contributed by Xiao Han in Tsinghua University.
## Datasets
- KGE.zip
## Supported Papers
- ManifoldE (IJCAI.2016): http://www.ibookman.net/IJCAI.2016.ManifoldE.pdf
- TransG (ACL.2016): http://www.ibookman.net/ACL.2016.TransG.pdf
- SSP (AAAI.2017): http://www.ibookman.net/AAAI.2017.SSP.pdf
- TransA (Arxiv): http://www.ibookman.net/Arxiv.TransA.pdf
- KSR (submitting to ACL.2017): http://www.ibookman.net/Arixv.KSR.pdf
## Citation
Conventionally, if this project helps you, please cite our paper, corresponddingly.
- Han Xiao, Minlie Huang, Xiaoyan Zhu. From One Point to A Manifold: Orbit Models for Knowledge Graph Embedding. The 25th International Joint Conference on Artificial Intelligence (IJCAI'16).
- Han Xiao, Minlie Huang, Xiaoyan Zhu. TransG: A Generative Mixture Model for Knowledge Graph Embedding. The 54th Annual Meeting of the Association for Computational Linguistics (ACL'2016).
- Han Xiao, Minlie Huang, Lian Meng, Xiaoyan Zhu. SSP: Semantic Space Projection for Knowledge Graph Embedding with Text Descriptions. The Thirty-First AAAI Conference on Artificial Intelligence (AAAI'17).
## BibTex
```
@article{Xiao2015From,
title={From One Point to A Manifold: Orbit Models for Knowledge Graph Embedding},
author={Xiao, Han and Huang, Minlie and Hao, Yu and Zhu, Xiaoyan},
journal={Computer Science},
year={2015},
}
@article{Xiao2016TransG,
title={TransG : A Generative Mixture Model for Knowledge Graph Embedding},
author={Xiao, Han and Huang, Minlie and Hao, Yu and Zhu, Xiaoyan},
journal={Computer Science},
year={2016},
}
@article{Xiao2016SSP,
title={SSP: Semantic Space Projection for Knowledge Graph Embedding with Text Descriptions},
author={Xiao, Han and Huang, Minlie and Zhu, Xiaoyan},
year={2016},
}
```
## Dependency
- Armadillo
- Armadillo is a high quality linear algebra library (matrix maths) for the C++ language, aiming towards a good balance between speed and ease of use
- I bet you could master it, just by scanning the examples.
- Download URL: [http://arma.sourceforge.net/download.html](http://arma.sourceforge.net/download.html "http://arma.sourceforge.net/download.html")
- What all you should do is to copy the headers into your environment.
- Boost
- C++ Standard Extensive Library.
- Download URL:[http://www.boost.org/users/download/](http://www.boost.org/users/download/ "http://www.boost.org/users/download/")
- What all you should do is to copy the headers into your environment. Certainly, you could compile the code just as explained in the website.
- MKL
- **Not Necessary**, but I strongly suggest you could take advantage of your devices.
## Basic Configuration
- Windows
- This project is naturally built on Visual Studio 2013 with Intel C++ Compiler 2016. If we share the same development perference, I guess you could start your work, right now.
- When you decide to compile it with MSC, there is a little trouble, because you shoud adjust your configuration.
- Linux / MAC
- I also apply the Intel C++ Compiler, which could be substituted by GCC, theoretically.
- `icc -std=c++11 -O3 -xHost -qopenmp -m32 Embedding.cpp`
## Start
- To justify your data source, please modify the `MultiChannelEmbedding\DetailedConfig.hpp`.
- To explore the correspondding method, just fill the template in `MultiChannelEmbedding\Embedding.cpp` with hyper-parameters.
- ` model = new MFactorE(FB15K, LinkPredictionTail, report_path, 10, 0.01, 0.1, 0.01, 10);`
- ` model->run(10000);`
- ` model->test();`
- ` delete model;`
- **Notably, our code needs a little more turns to converge, we suggest 10,000 rounds for each experiment.** This is a critical trick for repeating our experiments.
## Alias
- OrbitE = ManifoldE
- MFactorE = KSR
## Code Structures
- Import.hpp imports the headers of project.
- DataModel.hpp specifies the data structure of knowledge graph.
- Model.hpp specifies the training and testing process of knowledge embedding model.
- DetailedConfig.hpp specifies the detailed configuration of project, such as file path.
- Embedding.cpp is the main source file.
- GeometricModel.hpp implements TransE, TransH and TransR.
- OrbitModel.hpp implements ManifoldE.
- SemanticModel.hpp implements SSP.
- LatentModel.hpp implements KSR.