QMWF-LM

This is the code of the paper:

A Quantum Many-body wave function Inspired Language Modeling Approach

DEPENDENCIES

• python 2.7+
• numpy
• tensorflow 1.2+
• scikit-learn (sklearn)
• pandas

This model is for TREC-QA dataset and WIKI-QA dataset. the pretrained embedding we used is glove

RUN

You can run this model by:

python train.py --data trec --clean False 

python train.py --data wiki --clean True

REFERENCES

If you find this code is useful, please consider citing our work.

@inproceedings{zhang2018quantum,
  title={A quantum many-body wave function inspired language modeling approach},
  author={Zhang, Peng and Su, Zhan and Zhang, Lipeng and Wang, Benyou and Song, Dawei},
  booktitle={Proceedings of the 27th ACM International Conference on Information and Knowledge Management},
  pages={1303--1312},
  year={2018},
  organization={ACM}
}

Contributor

• @ZhanSu