Source code of "Socialformer: Social Network Inspired Long Document Modeling for Document Ranking".
- For double-blind review, we use anonymous github to hide the information that may reveal the identity.
- We would like to express our deep appreciation for all reviewers.
- python >= 3.6.0
- pytorch >= 1.9.0 (with GPU support)
- Transformers >= 4.5.1
- pytrec-eval == 0.5
- Referring to
/dataprocess
, including probability computing, graph construction, and graph partition.
bash ./dataprocess/run.sh
- Referring to
/model
, train and test the model.
bash ./model/run.sh