SNUH

The PyTorch (1.2.0) implementation of "Integrating Semantics and Neighborhood Information with Graph-Driven Generative Models for Document Retrieval" (ACL 2021).

The MindSpore implementation is available in this repo.

Datasets

We follow the setting of VDSH (Chaidaroon and Fang, 2017). Please download the data from here and move them into the ./data/ directory.

Quick start

Unsupervised document hashing on 20Newsgroups using 64 bits

python main.py ng64 data/ng20.tfidf.mat --train --cuda

To reproduce the results reported in the paper, please refer to the run.sh for detailed running commands.

Acknowledgement

The coding logic follows the project organization in AMMI.

License

This code is offered under the MIT License.

References

[1] Integrating Semantics and Neighborhood Information with Graph-Driven Generative Models for Document Retrieval

@article{zijing2021snuh,
  author    = {Zijing Ou and
               Qinliang Su and
               Jianxing Yu and
               Bang Liu and
               Jingwen Wang and
               Ruihui Zhao and
               Changyou Chen and
               Yefeng Zheng},
  title     = {Integrating Semantics and Neighborhood Information with Graph-Driven
               Generative Models for Document Retrieval},
  journal   = {arXiv preprint arXiv:2105.13066},
  year      = {2021},
}