NQE

KO | EN

gitlite — search

#javascript #hacktoberfest #llm #go #machine-learning #deep-learning #macos #java #cli #kubernetes #awesome-list #security #openai #mcp

(★ 24)

[AAAI 2023] Official resources of "NQE: N-ary Query Embedding for Complex Query Answering over Hyper-relational Knowledge Graphs".

NQE 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# NQE

Official resources of **"NQE: N-ary Query Embedding for Complex Query Answering over Hyper-Relational Knowledge Graphs"**. Haoran Luo, Haihong E, Yuhao Yang, Gengxian Zhou, Yikai Guo, Tianyu Yao, Zichen Tang, Xueyuan Lin, Kaiyang Wan. **AAAI 2023** \[[paper](https://doi.org/10.1609/aaai.v37i4.25576)\].

## Overview
An example of n-ary FOL query:
![](https://raw.githubusercontent.com/LHRLAB/NQE/main/figs/F1.drawio.png)

16 kinds of n-ary FOL queries in WD50K-NFOL:
![](https://raw.githubusercontent.com/LHRLAB/NQE/main/figs/F4.drawio.png)

## Setup

### Default implementation environment

* *Linux(SSH) + Python3.7.13 + Pytorch1.8.1 + Cuda10.2*

```
pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
```

### Install Dependencies

Install dependencies:
```
pip install -r requirements.txt
```

### Configure the Dataset

We tested the effectiveness of our model on two datasets, including the WD50K-QE dataset and the WD50K-NFOL dataset.

* WD50K-QE dataset is a dataset created with the multi-hop reasoning method StarQE. It covering multi-hop reasoning with conjunction logic operation. We call it "wd50k_qe" in the code. You can download and upzip the WD50K-QE dataset file from [TeraBox](https://1024terabox.com/s/1ZteuaNk2oq0VN17VccQt7w) or [Baidu Netdisk](https://pan.baidu.com/s/1IIFaPfXQIKKdWbX1HX6heA?pwd=26mp).

```
unzip wd50k_qe.zip -d data/
```

* WD50K-NFOL is a hyper-relational dataset we created covering logical operations including conjunction, disjunction and negation as well as combined operations. We call it "wd50k_nfol" in the code. You can download and upzip the WD50K-NFOL dataset file from [TeraBox](https://1024terabox.com/s/1EjkByMr1VqIcEbw6CClsag) or [Baidu Netdisk](https://pan.baidu.com/s/1D8ATQIEuwlZIa3LJJRjRPg?pwd=72ag).

```
unzip wd50k_nfol.zip -d data/
```

### Generate the Groundtruth ###

Then, we should generate the groundtruth of the chosen dataset for evaluation. If you don't change the dataset, please skip this step, because the zip files above have already got the groundtruth in "gt\" file by following operation.

* For WD50K-QE dataset:
```
python src/generate_groundtruth.py --dataset wd50k_qe
```
* For WD50K-NFOL dataset:
```
python src/generate_groundtruth.py --dataset wd50k_nfol
```

## Model Training ##

You can train query embedding model using "src/map_iter_qe.py".

* For WD50K-QE dataset:
```
python src/map_iter_qe.py --dataset wd50k_qe --epoch 300 --gpu_index 0
```

* For WD50K-NFOL dataset:
```
python src/map_iter_qe.py --dataset wd50k_nfol --epoch 50 --gpu_index 0
```

## Evaluation ##

After training, you can only run prediction using "src/map_iter_qe.py" by with argument "do_learn" False and argument "do_predict" True. In this case, you need to select the ckpts file you want to use and configure the "prediction_ckpt" argument as you want.

* For WD50K-QE dataset:
```
python src/map_iter_qe.py --dataset wd50k_qe --do_learn False --do_predict True --prediction_ckpt ckptswd50k_qe-train_tasks-1p-best-valid-DIM256.ckpt --prediction_tasks 1p,2p,3p,2i,3i,pi,ip --gpu_index 0
```

* For WD50K-NFOL dataset:
```
python src/map_iter_qe.py --dataset wd50k_nfol --do_learn False --do_predict True --prediction_ckpt ckptswd50k_nfol-train_tasks-1p-best-valid-DIM256.ckpt --prediction_tasks 1p,2p,3p,2i,3i,pi,ip,2u,up,2cp,3cp,2in,3in,inp,pin,pni --gpu_index 0
```

## BibTex

If you find this work is helpful for your research, please cite:

```bibtex
@article{luo2023nqe, 
  title={NQE: N-ary Query Embedding for Complex Query Answering over Hyper-Relational Knowledge Graphs}, 
  volume={37}, 
  url={https://ojs.aaai.org/index.php/AAAI/article/view/25576}, 
  DOI={10.1609/aaai.v37i4.25576}, 
  author={Luo, Haoran and E, Haihong and Yang, Yuhao and Zhou, Gengxian and Guo, Yikai and Yao, Tianyu and Tang, Zichen and Lin, Xueyuan and Wan, Kaiyang}, 
  year={2023}, 
  month={Jun.}, 
  pages={4543-4551} 
}
```

For further questions, please contact: haoran.luo@ieee.org.

NQE

NQE 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

NQE

NQE 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README