| --- |
| license: cc-by-sa-4.0 |
| language: |
| - en |
| base_model: distilroberta-base |
| pipeline_tag: text-classification |
| datasets: |
| - KIZZPER/pandas-so-questions |
| tags: |
| - pandas |
| - stackoverflow |
| - retrieval |
| --- |
| |
| # Классификатор вопросов про pandas по функциям |
|
|
| distilroberta-base, классы — 691 функция pandas (встречались в train не реже 3 раз). |
| Метки классов — имена страниц API Reference (`pandas.DataFrame.merge`). Обучена с soft-target cross-entropy: |
| если ответ ссылается на несколько функций, вероятность делится между ними поровну. |
|
|
| ## Качество на test (вопросов: 6 742) |
| | Recall@1 | Recall@5 | Recall@10 | MRR | |
| |---|---|---|---| |
| | 0.247 | 0.519 | 0.639 | 0.373 | |
|
|
| В демо проекта используется в смеси с `KIZZPER/pandas-so-function-retriever`. |
|
|
| Код, эксперименты и отчёты: репозиторий проекта (см. README). |
|
|
| Обучено на вопросах и ответах StackOverflow, которые распространяются по лицензии [CC BY-SA 4.0](https://creativecommons.org/licenses/by-sa/4.0/). Источник данных: [Stack Exchange Data Explorer](https://data.stackexchange.com/stackoverflow). |
|
|
