├── CL.py ├── GCN.py ├── README.md ├── Trainer.py ├── data ├── mr_data │ ├── adj_query2entity.pkl │ ├── adj_query2tag.pkl │ ├── adj_query2word.pkl │ ├── adj_tag.pkl │ ├── adj_word.pkl │ ├── entity_emb.pkl │ ├── labels.json │ ├── test_idx.json │ ├── train_idx.json │ └── word_emb.pkl └── snippets_data │ ├── adj_query2entity.pkl │ ├── adj_query2tag.pkl │ ├── adj_query2word.pkl │ ├── adj_tag.pkl │ ├── adj_word.pkl │ ├── entity_emb.pkl │ ├── labels.json │ ├── test_idx.json │ ├── train_idx.json │ └── word_emb.pkl ├── model.py ├── original_datasets ├── MR │ ├── mr_idx_split.json │ └── mr_split.json ├── Snippets │ ├── snippets_idx_split.json │ └── snippets_split.json ├── StackOverflow │ ├── StackOverflow_idx_split.json │ └── StackOverflow_split.json └── Twitter │ ├── twitter_idx_split.json │ └── twitter_split.json ├── preprocess.py ├── train.py └── utils.py /CL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/CL.py -------------------------------------------------------------------------------- /GCN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/GCN.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/README.md -------------------------------------------------------------------------------- /Trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/Trainer.py -------------------------------------------------------------------------------- /data/mr_data/adj_query2entity.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/mr_data/adj_query2entity.pkl -------------------------------------------------------------------------------- /data/mr_data/adj_query2tag.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/mr_data/adj_query2tag.pkl -------------------------------------------------------------------------------- /data/mr_data/adj_query2word.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/mr_data/adj_query2word.pkl -------------------------------------------------------------------------------- /data/mr_data/adj_tag.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/mr_data/adj_tag.pkl -------------------------------------------------------------------------------- /data/mr_data/adj_word.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/mr_data/adj_word.pkl -------------------------------------------------------------------------------- /data/mr_data/entity_emb.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/mr_data/entity_emb.pkl -------------------------------------------------------------------------------- /data/mr_data/labels.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/mr_data/labels.json -------------------------------------------------------------------------------- /data/mr_data/test_idx.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/mr_data/test_idx.json -------------------------------------------------------------------------------- /data/mr_data/train_idx.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/mr_data/train_idx.json -------------------------------------------------------------------------------- /data/mr_data/word_emb.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/mr_data/word_emb.pkl -------------------------------------------------------------------------------- /data/snippets_data/adj_query2entity.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/snippets_data/adj_query2entity.pkl -------------------------------------------------------------------------------- /data/snippets_data/adj_query2tag.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/snippets_data/adj_query2tag.pkl -------------------------------------------------------------------------------- /data/snippets_data/adj_query2word.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/snippets_data/adj_query2word.pkl -------------------------------------------------------------------------------- /data/snippets_data/adj_tag.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/snippets_data/adj_tag.pkl -------------------------------------------------------------------------------- /data/snippets_data/adj_word.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/snippets_data/adj_word.pkl -------------------------------------------------------------------------------- /data/snippets_data/entity_emb.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/snippets_data/entity_emb.pkl -------------------------------------------------------------------------------- /data/snippets_data/labels.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/snippets_data/labels.json -------------------------------------------------------------------------------- /data/snippets_data/test_idx.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/snippets_data/test_idx.json -------------------------------------------------------------------------------- /data/snippets_data/train_idx.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/snippets_data/train_idx.json -------------------------------------------------------------------------------- /data/snippets_data/word_emb.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/data/snippets_data/word_emb.pkl -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/model.py -------------------------------------------------------------------------------- /original_datasets/MR/mr_idx_split.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/original_datasets/MR/mr_idx_split.json -------------------------------------------------------------------------------- /original_datasets/MR/mr_split.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/original_datasets/MR/mr_split.json -------------------------------------------------------------------------------- /original_datasets/Snippets/snippets_idx_split.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/original_datasets/Snippets/snippets_idx_split.json -------------------------------------------------------------------------------- /original_datasets/Snippets/snippets_split.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/original_datasets/Snippets/snippets_split.json -------------------------------------------------------------------------------- /original_datasets/StackOverflow/StackOverflow_idx_split.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/original_datasets/StackOverflow/StackOverflow_idx_split.json -------------------------------------------------------------------------------- /original_datasets/StackOverflow/StackOverflow_split.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/original_datasets/StackOverflow/StackOverflow_split.json -------------------------------------------------------------------------------- /original_datasets/Twitter/twitter_idx_split.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/original_datasets/Twitter/twitter_idx_split.json -------------------------------------------------------------------------------- /original_datasets/Twitter/twitter_split.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/original_datasets/Twitter/twitter_split.json -------------------------------------------------------------------------------- /preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/preprocess.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KEAML-JLU/SimSTC/HEAD/utils.py --------------------------------------------------------------------------------