├── README.md ├── config.yaml ├── datasets ├── ToolBenchG2 │ ├── ToolBenchG2_data_size.txt │ ├── corpus.jsonl │ ├── qrels │ │ ├── test.tsv │ │ └── train.tsv │ ├── queries.jsonl │ ├── query_scene_test.txt │ ├── query_scene_train.txt │ ├── query_scene_tune.txt │ ├── query_tool.txt │ ├── query_tool_test.txt │ ├── query_tool_train.txt │ ├── query_tool_tune.txt │ └── scene_tool.txt ├── ToolBenchG3 │ ├── ToolBenchG3_data_size.txt │ ├── corpus.jsonl │ ├── qrels │ │ ├── test.tsv │ │ └── train.tsv │ ├── queries.jsonl │ ├── query_scene_test.txt │ ├── query_scene_train.txt │ ├── query_scene_tune.txt │ ├── query_tool.txt │ ├── query_tool_test.txt │ ├── query_tool_train.txt │ ├── query_tool_tune.txt │ └── scene_tool.txt └── ToolLens │ ├── ToolLens_data.json │ ├── ToolLens_data_size.txt │ ├── corpus.jsonl │ ├── qrels │ ├── test.tsv │ └── train.tsv │ ├── queries.jsonl │ ├── query_scene_test.txt │ ├── query_scene_train.txt │ ├── query_scene_tune.txt │ ├── query_tool.txt │ ├── query_tool_test.txt │ ├── query_tool_train.txt │ ├── query_tool_tune.txt │ └── scene_tool.txt ├── models └── COLT.py ├── run.sh ├── train.py ├── train_sbert.py └── utility.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/README.md -------------------------------------------------------------------------------- /config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/config.yaml -------------------------------------------------------------------------------- /datasets/ToolBenchG2/ToolBenchG2_data_size.txt: -------------------------------------------------------------------------------- 1 | 82507 36306 11474 2 | -------------------------------------------------------------------------------- /datasets/ToolBenchG2/corpus.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG2/corpus.jsonl -------------------------------------------------------------------------------- /datasets/ToolBenchG2/qrels/test.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG2/qrels/test.tsv -------------------------------------------------------------------------------- /datasets/ToolBenchG2/qrels/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG2/qrels/train.tsv -------------------------------------------------------------------------------- /datasets/ToolBenchG2/queries.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG2/queries.jsonl -------------------------------------------------------------------------------- /datasets/ToolBenchG2/query_scene_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG2/query_scene_test.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG2/query_scene_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG2/query_scene_train.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG2/query_scene_tune.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG2/query_scene_tune.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG2/query_tool.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG2/query_tool.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG2/query_tool_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG2/query_tool_test.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG2/query_tool_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG2/query_tool_train.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG2/query_tool_tune.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG2/query_tool_tune.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG2/scene_tool.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG2/scene_tool.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG3/ToolBenchG3_data_size.txt: -------------------------------------------------------------------------------- 1 | 23734 9272 1419 2 | -------------------------------------------------------------------------------- /datasets/ToolBenchG3/corpus.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG3/corpus.jsonl -------------------------------------------------------------------------------- /datasets/ToolBenchG3/qrels/test.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG3/qrels/test.tsv -------------------------------------------------------------------------------- /datasets/ToolBenchG3/qrels/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG3/qrels/train.tsv -------------------------------------------------------------------------------- /datasets/ToolBenchG3/queries.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG3/queries.jsonl -------------------------------------------------------------------------------- /datasets/ToolBenchG3/query_scene_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG3/query_scene_test.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG3/query_scene_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG3/query_scene_train.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG3/query_scene_tune.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG3/query_scene_tune.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG3/query_tool.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG3/query_tool.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG3/query_tool_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG3/query_tool_test.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG3/query_tool_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG3/query_tool_train.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG3/query_tool_tune.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG3/query_tool_tune.txt -------------------------------------------------------------------------------- /datasets/ToolBenchG3/scene_tool.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolBenchG3/scene_tool.txt -------------------------------------------------------------------------------- /datasets/ToolLens/ToolLens_data.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolLens/ToolLens_data.json -------------------------------------------------------------------------------- /datasets/ToolLens/ToolLens_data_size.txt: -------------------------------------------------------------------------------- 1 | 18770 463 464 2 | -------------------------------------------------------------------------------- /datasets/ToolLens/corpus.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolLens/corpus.jsonl -------------------------------------------------------------------------------- /datasets/ToolLens/qrels/test.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolLens/qrels/test.tsv -------------------------------------------------------------------------------- /datasets/ToolLens/qrels/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolLens/qrels/train.tsv -------------------------------------------------------------------------------- /datasets/ToolLens/queries.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolLens/queries.jsonl -------------------------------------------------------------------------------- /datasets/ToolLens/query_scene_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolLens/query_scene_test.txt -------------------------------------------------------------------------------- /datasets/ToolLens/query_scene_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolLens/query_scene_train.txt -------------------------------------------------------------------------------- /datasets/ToolLens/query_scene_tune.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolLens/query_scene_tune.txt -------------------------------------------------------------------------------- /datasets/ToolLens/query_tool.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolLens/query_tool.txt -------------------------------------------------------------------------------- /datasets/ToolLens/query_tool_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolLens/query_tool_test.txt -------------------------------------------------------------------------------- /datasets/ToolLens/query_tool_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolLens/query_tool_train.txt -------------------------------------------------------------------------------- /datasets/ToolLens/query_tool_tune.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolLens/query_tool_tune.txt -------------------------------------------------------------------------------- /datasets/ToolLens/scene_tool.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/datasets/ToolLens/scene_tool.txt -------------------------------------------------------------------------------- /models/COLT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/models/COLT.py -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/run.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/train.py -------------------------------------------------------------------------------- /train_sbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/train_sbert.py -------------------------------------------------------------------------------- /utility.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quchangle1/COLT/HEAD/utility.py --------------------------------------------------------------------------------