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

Fix arguments error in calculating ndcg_score for retrieval model #266

Open neneluo opened 2 months ago

neneluo commented 2 months ago

Hello,

Thank you for open-sourcing your code! :) I encountered an error while training the retrieval model, which indicated a mismatch in the number of arguments when calculating metrics. It turned out to be because the 'k' argument was not explicitly specified in the computation of ndcg_score.

Hope this helps. #208