KIZZPER/pandas-so-function-retriever

Поиск функций pandas по вопросу (bi-encoder)

bge-small-en-v1.5, дообученная контрастивным loss на 45 тыс. парах «вопрос StackOverflow -> функция pandas, на которую ссылается ответ». Пулинг по CLS, эмбеддинги нормированы, близость — скалярное произведение.

api_corpus.parquet — 2 019 страниц API Reference pandas 3.0.6 (поле text кодируется как документ, url — проверенная ссылка на страницу документации).

Качество на test (вопросов: 6 742)

	Recall@1	Recall@5	Recall@10	MRR
только эта модель	0.245	0.510	0.633	0.373
смесь с KIZZPER/pandas-so-function-classifier (0.4 / 0.6)	0.280	0.578	0.698	0.418

Код, эксперименты и отчёты: репозиторий проекта (см. README).

Обучено на вопросах и ответах StackOverflow, которые распространяются по лицензии CC BY-SA 4.0. Источник данных: Stack Exchange Data Explorer.