OpenBMB / ToolBench

[ICLR'24 spotlight] An open platform for training, serving, and evaluating large language model for tool learning.
https://openbmb.github.io/ToolBench/
Apache License 2.0
4.6k stars 397 forks source link

all training data for API Retriever have label 1 #241

Open ZongqianLi opened 4 months ago

ZongqianLi commented 4 months ago

Thank you for sharing the codes and data. This is really an amazong work!

In page 8 of the paper, it says: "During training, we regard the relevant APIs of each instruction generated in 2.2 as positive examples and also sample a few APIs as negative examples for contrastive learning" However, in toolbench/retrieval/train.py, train_samples only have one label value, 1. Could I ask whether I have missed the contrastive learning or it is not in the codes?